bugfix: fix params
Browse files- inference.py +2 -2
inference.py
CHANGED
@@ -164,7 +164,7 @@ class DiffusionInference:
|
|
164 |
print(f"Warning: Could not delete temporary file {temp_file}: {e}")
|
165 |
|
166 |
@spaces.GPU
|
167 |
-
def run_text_to_image_pipeline(self,
|
168 |
-
pipeline = AutoPipelineForText2Image.from_pretrained(model=
|
169 |
image = pipeline(**kwargs).images[0]
|
170 |
return image
|
|
|
164 |
print(f"Warning: Could not delete temporary file {temp_file}: {e}")
|
165 |
|
166 |
@spaces.GPU
|
167 |
+
def run_text_to_image_pipeline(self, **kwargs):
|
168 |
+
pipeline = AutoPipelineForText2Image.from_pretrained(model=model, torch_dtype=torch.float16).to("cuda")
|
169 |
image = pipeline(**kwargs).images[0]
|
170 |
return image
|