ktrndy commited on
Commit
d9bbac9
·
verified ·
1 Parent(s): 909eb62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from peft import PeftModel, LoraConfig
8
  from diffusers import DiffusionPipeline
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
- model_id_default = "CompVis/stable-diffusion-v1-4" # Replace to the model you would like to use
12
 
13
  if torch.cuda.is_available():
14
  torch_dtype = torch.float16
 
8
  from diffusers import DiffusionPipeline
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
+ model_id_default = "stable-diffusion-v1-5/stable-diffusion-v1-5" # Replace to the model you would like to use
12
 
13
  if torch.cuda.is_available():
14
  torch_dtype = torch.float16