lvkaokao commited on
Commit
31342b3
1 Parent(s): 7d2e067
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -8,12 +8,10 @@ import subprocess
8
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
9
  import time
10
 
11
- repo_id = "runwayml/stable-diffusion-v1-5"
12
-
13
- dpm = DPMSolverMultistepScheduler.from_pretrained(repo_id, subfolder="scheduler")
14
-
15
  model_id = "dicoo_model"
16
 
 
 
17
  pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=dpm, torch_dtype=torch.float)
18
 
19
  def predict(prompt, steps=25, seed=42, guidance_scale=7.5):
 
8
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
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):