Spaces:
Paused
Paused
if prompt_embeds is None:
Browse files
latent_consistency_img2img.py
CHANGED
@@ -118,11 +118,11 @@ class LatentConsistencyModelImg2ImgPipeline(DiffusionPipeline):
|
|
118 |
prompt,
|
119 |
padding="max_length",
|
120 |
max_length=self.tokenizer.model_max_length,
|
121 |
-
truncation=
|
122 |
return_tensors="pt",
|
123 |
)
|
124 |
|
125 |
-
text_input_ids = text_inputs.input_ids
|
126 |
untruncated_ids = self.tokenizer(
|
127 |
prompt, padding="longest", return_tensors="pt"
|
128 |
).input_ids
|
|
|
118 |
prompt,
|
119 |
padding="max_length",
|
120 |
max_length=self.tokenizer.model_max_length,
|
121 |
+
truncation=True,
|
122 |
return_tensors="pt",
|
123 |
)
|
124 |
|
125 |
+
text_input_ids = text_inputs.input_ids
|
126 |
untruncated_ids = self.tokenizer(
|
127 |
prompt, padding="longest", return_tensors="pt"
|
128 |
).input_ids
|