fffiloni commited on
Commit
76ed668
1 Parent(s): 58ce576

Change MAX_IMAGE_SIZE to 16384

Browse files

Change MAX_IMAGE_SIZE to 16384 to reflect the paper claim :)

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ else:
20
  torch_dtype = torch.float32
21
 
22
  MAX_SEED = np.iinfo(np.int32).max
23
- MAX_IMAGE_SIZE = 1024
24
  pipes = {}
25
  for model_id, repo_id in all_model_id.items():
26
  pipes[model_id + '_t2i'] = StableDiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)
 
20
  torch_dtype = torch.float32
21
 
22
  MAX_SEED = np.iinfo(np.int32).max
23
+ MAX_IMAGE_SIZE = 16384
24
  pipes = {}
25
  for model_id, repo_id in all_model_id.items():
26
  pipes[model_id + '_t2i'] = StableDiffusionPipeline.from_pretrained(repo_id, torch_dtype=torch_dtype)