Kotomiya07
commited on
Commit
•
fc8c2a5
1
Parent(s):
28b36fe
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
|
|
23 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
|
24 |
MIN_IMAGE_SIZE = int(os.getenv("MIN_IMAGE_SIZE", "512"))
|
25 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "2048"))
|
26 |
-
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
|
27 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
|
28 |
OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
|
29 |
|
|
|
23 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES") == "1"
|
24 |
MIN_IMAGE_SIZE = int(os.getenv("MIN_IMAGE_SIZE", "512"))
|
25 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "2048"))
|
26 |
+
USE_TORCH_COMPILE = False #os.getenv("USE_TORCH_COMPILE") == "1"
|
27 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
|
28 |
OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
|
29 |
|