respect --outdir again; fix issue #628

This commit is contained in:
Lincoln Stein
2022-09-16 19:58:45 -04:00
parent df95a7ddf2
commit df4c80f177
10 changed files with 206 additions and 158 deletions

View File

@@ -82,7 +82,9 @@ class EmbeddingManager(nn.Module):
get_embedding_for_clip_token,
embedder.transformer.text_model.embeddings,
)
token_dim = 1280
# per bug report #572
#token_dim = 1280
token_dim = 768
else: # using LDM's BERT encoder
self.is_clip = False
get_token_for_string = partial(