lichih commited on
Commit
4d0bc77
·
verified ·
1 Parent(s): 7a0dabc

change model source

Browse files

fix the issue that `ruwnayml/stable-diffusion-v1-5` is deprecated.

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,11 +9,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
  if torch.cuda.is_available():
11
  torch.cuda.max_memory_allocated(device=device)
12
- pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
13
  #pipe.enable_xformers_memory_efficient_attention()
14
  pipe = pipe.to(device)
15
  else:
16
- pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
17
  pipe = pipe.to(device)
18
 
19
  MAX_SEED = np.iinfo(np.int32).max
 
9
 
10
  if torch.cuda.is_available():
11
  torch.cuda.max_memory_allocated(device=device)
12
+ pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16)
13
  #pipe.enable_xformers_memory_efficient_attention()
14
  pipe = pipe.to(device)
15
  else:
16
+ pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5")
17
  pipe = pipe.to(device)
18
 
19
  MAX_SEED = np.iinfo(np.int32).max