Spaces:
Running on Zero
Running on Zero
App: fix paths.
Browse files
app.py
CHANGED
|
@@ -7,9 +7,9 @@ os.environ["TORCH_COMPILE_DISABLE"] = "1"
|
|
| 7 |
os.environ["TORCHDYNAMO_DISABLE"] = "1"
|
| 8 |
|
| 9 |
# Place caches in persistent storage
|
| 10 |
-
os.environ["HF_HOME"] = /data/.huggingface
|
| 11 |
-
os.environ["HF_HUB_CACHE"] = /data/.cache/huggingface/hub
|
| 12 |
-
os.environ["HF_DATASETS_CACHE"] = /data/.cache/huggingface/datasets
|
| 13 |
|
| 14 |
# Install xformers for memory-efficient attention
|
| 15 |
subprocess.run([sys.executable, "-m", "pip", "install", "xformers==0.0.32.post2", "--no-build-isolation"], check=False)
|
|
|
|
| 7 |
os.environ["TORCHDYNAMO_DISABLE"] = "1"
|
| 8 |
|
| 9 |
# Place caches in persistent storage
|
| 10 |
+
os.environ["HF_HOME"] = "/data/.huggingface"
|
| 11 |
+
os.environ["HF_HUB_CACHE"] = "/data/.cache/huggingface/hub"
|
| 12 |
+
os.environ["HF_DATASETS_CACHE"] = "/data/.cache/huggingface/datasets"
|
| 13 |
|
| 14 |
# Install xformers for memory-efficient attention
|
| 15 |
subprocess.run([sys.executable, "-m", "pip", "install", "xformers==0.0.32.post2", "--no-build-isolation"], check=False)
|