fix typo in imagic_stable_diffusion.py (huggingface#1956)

andreemic · web-flow · commit 4401e6aa2b94 · 2023-01-09T20:34:19.000+01:00
changed named arg to self.tokenizer from `truncaton` to `truncation` according to Tokenizer docs (https://huggingface.co/docs/tokenizers/api/tokenizer#tokenizers.Tokenizer.truncation)
diff --git a/examples/community/imagic_stable_diffusion.py b/examples/community/imagic_stable_diffusion.py
@@ -229,7 +229,7 @@ def train(
             prompt,
             padding="max_length",
             max_length=self.tokenizer.model_max_length,
-            truncaton=True,
+            truncation=True,
             return_tensors="pt",
         )
         text_embeddings = torch.nn.Parameter(

Original file line number	Diff line number	Diff line change
`@@ -229,7 +229,7 @@ def train(`
`229`	`229`	`prompt,`
`230`	`230`	`padding="max_length",`
`231`	`231`	`max_length=self.tokenizer.model_max_length,`
`232`		`- truncaton=True,`
	`232`	`+ truncation=True,`
`233`	`233`	`return_tensors="pt",`
`234`	`234`	`)`
`235`	`235`	`text_embeddings = torch.nn.Parameter(`