Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
d59a4dd
1
Parent(s):
c689a76
app.py
Browse files- app.py +1 -1
- requirements.txt +2 -1
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.float
|
| 40 |
|
| 41 |
LOW_MEMORY = os.getenv("LOW_MEMORY", "0") == "1"
|
| 42 |
|
requirements.txt
CHANGED
|
@@ -21,4 +21,5 @@ torch==2.2
|
|
| 21 |
controlnet-aux==0.0.9
|
| 22 |
onnx==1.16.1
|
| 23 |
onnxruntime==1.18.0
|
| 24 |
-
mediapipe==0.10.14
|
|
|
|
|
|
| 21 |
controlnet-aux==0.0.9
|
| 22 |
onnx==1.16.1
|
| 23 |
onnxruntime==1.18.0
|
| 24 |
+
mediapipe==0.10.14
|
| 25 |
+
peft==0.11.1
|