Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
97803a9
1
Parent(s):
d59a4dd
app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ IS_SPACES_ZERO = os.environ.get("SPACES_ZERO_GPU", "0") == "1"
|
|
| 36 |
IS_SPACE = os.environ.get("SPACE_ID", None) is not None
|
| 37 |
|
| 38 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 39 |
-
dtype = torch.
|
| 40 |
|
| 41 |
LOW_MEMORY = os.getenv("LOW_MEMORY", "0") == "1"
|
| 42 |
|
|
|
|
| 36 |
IS_SPACE = os.environ.get("SPACE_ID", None) is not None
|
| 37 |
|
| 38 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 39 |
+
dtype = torch.float16
|
| 40 |
|
| 41 |
LOW_MEMORY = os.getenv("LOW_MEMORY", "0") == "1"
|
| 42 |
|