Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ DESCRIPTIONx = """## STABLE HAMSTER
|
|
15 |
|
16 |
# Use environment variables for flexibility
|
17 |
MODEL_ID = os.getenv("MODEL_REPO")
|
18 |
-
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "
|
19 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
|
20 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
21 |
BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1")) # Allow generating multiple images at once
|
|
|
15 |
|
16 |
# Use environment variables for flexibility
|
17 |
MODEL_ID = os.getenv("MODEL_REPO")
|
18 |
+
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "2048"))
|
19 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
|
20 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
21 |
BATCH_SIZE = int(os.getenv("BATCH_SIZE", "1")) # Allow generating multiple images at once
|