Spaces:
Sleeping
Sleeping
Change MAX_IMAGE_SIZE to 16384
Browse filesChange MAX_IMAGE_SIZE to 16384 to reflect the paper claim :)
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 =
|
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)
|