Huage001 commited on
Commit
d03f62a
·
verified ·
1 Parent(s): d5406ce

Update app.py

Browse files
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 = 2048
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 = 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)