lvkaokao commited on
Commit
dbf5ce1
1 Parent(s): 31342b3

update app

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -10,9 +10,9 @@ import time
10
 
11
  model_id = "dicoo_model"
12
 
13
- dpm = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
14
-
15
- pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=dpm, torch_dtype=torch.float)
16
 
17
  def predict(prompt, steps=25, seed=42, guidance_scale=7.5):
18
  # cpu info
 
10
 
11
  model_id = "dicoo_model"
12
 
13
+ #dpm = DPMSolverMultistepScheduler.from_pretrained(model_id, subfolder="scheduler")
14
+ #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=dpm, torch_dtype=torch.float)
15
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float)
16
 
17
  def predict(prompt, steps=25, seed=42, guidance_scale=7.5):
18
  # cpu info