pipeline(

latent_consistency_img2img.py

@@ -118,10 +118,19 @@ class LatentConsistencyModelImg2ImgPipeline(DiffusionPipeline):
                 prompt,
                 padding="max_length",
                 max_length=self.tokenizer.model_max_length,
-                truncation=True,
+                truncation=False,
                 return_tensors="pt",
             )
-            text_input_ids = text_inputs.input_ids
+            negative_prompt = "ugly"
+            negative_ids = pipeline.tokenizer(
+                negative_prompt, 
+                truncation=False, 
+                padding="max_length",
+                max_length=input_ids.shape[-1], 
+                return_tensors="pt"
+            ).input_ids.to("cuda")
+
+            text_input_ids = text_inputs.input_ids("cuda")
             untruncated_ids = self.tokenizer(
                 prompt, padding="longest", return_tensors="pt"
             ).input_ids