ovedrive commited on
Commit
4e7d702
·
1 Parent(s): ce402ac

fix: seed to gnerator

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -163,7 +163,7 @@ class DiffusionInference:
163
  print(f"Warning: Could not delete temporary file {temp_file}: {e}")
164
 
165
  @spaces.GPU
166
- def run_text_to_image_pipeline(self, model_name, **kwargs):
167
  pipeline = AutoPipelineForText2Image.from_pretrained(model_name, generator=generator, torch_dtype=torch.float16).to("cuda")
168
  image = pipeline(**kwargs).images[0]
169
  return image
 
163
  print(f"Warning: Could not delete temporary file {temp_file}: {e}")
164
 
165
  @spaces.GPU
166
+ def run_text_to_image_pipeline(self, model_name, generator, **kwargs):
167
  pipeline = AutoPipelineForText2Image.from_pretrained(model_name, generator=generator, torch_dtype=torch.float16).to("cuda")
168
  image = pipeline(**kwargs).images[0]
169
  return image