lvkaokao commited on
Commit
224639d
1 Parent(s): dbf5ce1

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -10,9 +10,8 @@ import time
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
 
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
 
16
  def predict(prompt, steps=25, seed=42, guidance_scale=7.5):
17
  # cpu info