alexeyGod commited on
Commit
44c7044
·
verified ·
1 Parent(s): 54c09e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ login(token=a )
16
 
17
 
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
19
- model_repo_id = "black-forest-labs/FLUX.1-dev" # "stable-diffusion-v1-5/stable-diffusion-v1-5" # Replace to the model you would like to use
20
 
21
  if torch.cuda.is_available():
22
  torch_dtype = torch.float16
 
16
 
17
 
18
  device = "cuda" if torch.cuda.is_available() else "cpu"
19
+ model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo" # "stable-diffusion-v1-5/stable-diffusion-v1-5" # Replace to the model you would like to use
20
 
21
  if torch.cuda.is_available():
22
  torch_dtype = torch.float16