Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
4c27642
1
Parent(s):
47ed9da
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,7 +34,7 @@ model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.bfloa
|
|
| 34 |
model=model.to(device)
|
| 35 |
|
| 36 |
MAX_SEED = np.iinfo(np.int32).max
|
| 37 |
-
MAX_IMAGE_SIZE =
|
| 38 |
|
| 39 |
@spaces.GPU
|
| 40 |
def infer(seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
|
|
|
|
| 34 |
model=model.to(device)
|
| 35 |
|
| 36 |
MAX_SEED = np.iinfo(np.int32).max
|
| 37 |
+
MAX_IMAGE_SIZE = 1344
|
| 38 |
|
| 39 |
@spaces.GPU
|
| 40 |
def infer(seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
|