Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ if not torch.cuda.is_available():
|
|
44 |
MAX_SEED = np.iinfo(np.int32).max
|
45 |
CACHE_EXAMPLES = False
|
46 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
|
47 |
-
USE_TORCH_COMPILE =
|
48 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
49 |
|
50 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
|
44 |
MAX_SEED = np.iinfo(np.int32).max
|
45 |
CACHE_EXAMPLES = False
|
46 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1536"))
|
47 |
+
USE_TORCH_COMPILE = False
|
48 |
ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
|
49 |
|
50 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|