TheAIBoi commited on
Commit
7b743a3
·
verified ·
1 Parent(s): c0cc290

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,13 +12,13 @@ model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like
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)
18
  pipe = pipe.to(device)
19
 
20
  MAX_SEED = np.iinfo(np.int32).max
21
- MAX_IMAGE_SIZE = 1024
22
 
23
 
24
  # @spaces.GPU #[uncomment to use ZeroGPU]
 
12
  if torch.cuda.is_available():
13
  torch_dtype = torch.float16
14
  else:
15
+ torch_dtype = torch.float16
16
 
17
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
18
  pipe = pipe.to(device)
19
 
20
  MAX_SEED = np.iinfo(np.int32).max
21
+ MAX_IMAGE_SIZE = 4096
22
 
23
 
24
  # @spaces.GPU #[uncomment to use ZeroGPU]