Lookingsoft commited on
Commit
257bf5a
·
verified ·
1 Parent(s): d2191a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from spaces import GPU # 👈 only for HF Spaces
7
 
8
  login(token=os.getenv("HF_TOKEN"))
9
 
10
- model_id = "stabilityai/stable-diffusion-2-1-base"
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
 
13
  pipe = StableDiffusionPipeline.from_pretrained(
 
7
 
8
  login(token=os.getenv("HF_TOKEN"))
9
 
10
+ model_id = "runwayml/stable-diffusion-v1-5"
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
 
13
  pipe = StableDiffusionPipeline.from_pretrained(