Hugging Face's logo

Spaces:
alibaba-pai
/
pai-diffusion-artist-large-zh
Build error

App Files Community
2
chywang commited on
Commit
034660a
•
1 Parent(s): 2dfd9c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
3
  import torch
4
  from PIL import Image
5
 
6
- model_id = "alibaba-pai/pai-diffusion-general-large-zh"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id)
8
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
9
  pipe = pipe.to("cuda")
 
3
  import torch
4
  from PIL import Image
5
 
6
+ model_id = "alibaba-pai/pai-diffusion-artist-large-zh"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id)
8
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
9
  pipe = pipe.to("cuda")