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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ a=os.getenv('hf_key')
12
  login(token=a )
13
 
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
- model_repo_id = "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
  if torch.cuda.is_available():
18
  torch_dtype = torch.float16