SIGMitch commited on
Commit
b557de9
·
1 Parent(s): 90a12a3
Files changed (1) hide show
  1. latent_consistency_img2img.py +1 -1
latent_consistency_img2img.py CHANGED
@@ -122,7 +122,7 @@ class LatentConsistencyModelImg2ImgPipeline(DiffusionPipeline):
122
  return_tensors="pt",
123
  )
124
 
125
- text_input_ids = text_inputs.input_ids("cuda")
126
  untruncated_ids = self.tokenizer(
127
  prompt, padding="longest", return_tensors="pt"
128
  ).input_ids
 
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