AlanB
/

imagic_stable_diffusion_mod

Model card Files Files and versions

AlanB commited on Jan 10, 2023

Commit

4f94e1e

·

1 Parent(s): 5db5932

Fixed typo of truncation

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -253,7 +253,7 @@ class ImagicStableDiffusionPipeline(DiffusionPipeline):
             prompt,
             padding="max_length",
             max_length=self.tokenizer.model_max_length,
-            truncaton=True,
             return_tensors="pt",
         )
         text_embeddings = torch.nn.Parameter(

             prompt,
             padding="max_length",
             max_length=self.tokenizer.model_max_length,
+            truncation=True,
             return_tensors="pt",
         )
         text_embeddings = torch.nn.Parameter(