alexeyGod commited on
Commit
7599e27
·
verified ·
1 Parent(s): 9bfb47e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -12,7 +12,10 @@ a=os.getenv('hf_key')
12
  login(token=a )
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
- model_repo_id = "CompVis/stable-diffusion-v1-4"#"stabilityai/sdxl-turbo" # Replace to the model you would like to use
 
 
 
16
 
17
  if torch.cuda.is_available():
18
  torch_dtype = torch.float16
 
12
  login(token=a )
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
+ #model_repo_id = "CompVis/stable-diffusion-v1-4"#"stabilityai/sdxl-turbo" # Replace to the model you would like to use
16
+
17
+ model_repo_id = "chenguolin/stable-diffusion-v1-5"
18
+
19
 
20
  if torch.cuda.is_available():
21
  torch_dtype = torch.float16