Unverified Commit 4401e6aa authored by andreemic's avatar andreemic Committed by GitHub
Browse files

fix typo in imagic_stable_diffusion.py (#1956)

changed named arg to self.tokenizer from `truncaton` to `truncation` according to Tokenizer docs (https://huggingface.co/docs/tokenizers/api/tokenizer#tokenizers.Tokenizer.truncation)
parent 089f0f4c
...@@ -229,7 +229,7 @@ class ImagicStableDiffusionPipeline(DiffusionPipeline): ...@@ -229,7 +229,7 @@ class ImagicStableDiffusionPipeline(DiffusionPipeline):
prompt, prompt,
padding="max_length", padding="max_length",
max_length=self.tokenizer.model_max_length, max_length=self.tokenizer.model_max_length,
truncaton=True, truncation=True,
return_tensors="pt", return_tensors="pt",
) )
text_embeddings = torch.nn.Parameter( text_embeddings = torch.nn.Parameter(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment