Spaces:

lvkaokao
/

INC-Dicoo-Diffusion

Runtime error

lvkaokao commited on Nov 24, 2022

Commit

6bbf751

1 Parent(s): 414eb06

update app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,8 @@ model_id = "dicoo_model"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float)
 def predict(prompt, steps=50, seed=42, guidance_scale=7.5):
-    print("prompt:", prompt)
     image = pipe(prompt, num_inference_steps=steps, guidance_scale=7.5).images[0]
     return image

 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float)
 def predict(prompt, steps=50, seed=42, guidance_scale=7.5):
+    print("prompt: ", prompt)
+    print("steps: ", steps)
     image = pipe(prompt, num_inference_steps=steps, guidance_scale=7.5).images[0]
     return image