mortal99 commited on
Commit
55bd413
1 Parent(s): ced5105

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ with gr.Blocks() as demo:
28
  # 6.加载模型
29
  pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
30
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
31
- pipe.unet.load_attn_procs("model_output/", from_hf_hub=False)
32
 
33
  # 7.启动
34
  demo.launch()
 
28
  # 6.加载模型
29
  pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
30
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
31
+ pipe.unet.load_attn_procs("model_output/", from_hf_hub=True)
32
 
33
  # 7.启动
34
  demo.launch()