ovedrive commited on
Commit
aff01bb
·
1 Parent(s): 527eee1

bugfix: fix params

Browse files
Files changed (1) hide show
  1. 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,model_name, **kwargs):
168
- pipeline = AutoPipelineForText2Image.from_pretrained(model=model_name, torch_dtype=torch.float16).to("cuda")
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