fix: seed to gnerator
Browse files- 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
|