LEIDIA commited on
Commit
9eaff55
Β·
verified Β·
1 Parent(s): 018e6a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like
11
 
12
  if torch.cuda.is_available():
13
  torch_dtype = torch.float16
14
- else:
15
  torch_dtype = torch.float32
16
 
17
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 
11
 
12
  if torch.cuda.is_available():
13
  torch_dtype = torch.float16
14
+ else:
15
  torch_dtype = torch.float32
16
 
17
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)