Spaces:
Running on Zero
Running on Zero
second try at consistent weights
Browse files
app.py
CHANGED
|
@@ -13,11 +13,11 @@ from rfd3.engine import RFD3InferenceConfig, RFD3InferenceEngine
|
|
| 13 |
# Download model weights (skips already-downloaded models automatically)
|
| 14 |
# In total, ~6GB (3GB for RFD3, 3GB for RF3, <100MB for MPNN); may take a few minutes depending on your connection speed
|
| 15 |
|
| 16 |
-
cmd = "foundry install rfd3 ligandmpnn rf3"
|
| 17 |
-
print("Global PATH:", os.environ.get("PATH"))
|
| 18 |
|
| 19 |
-
|
| 20 |
-
|
|
|
|
|
|
|
| 21 |
|
| 22 |
result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
|
| 23 |
if result.returncode == 0:
|
|
|
|
| 13 |
# Download model weights (skips already-downloaded models automatically)
|
| 14 |
# In total, ~6GB (3GB for RFD3, 3GB for RF3, <100MB for MPNN); may take a few minutes depending on your connection speed
|
| 15 |
|
|
|
|
|
|
|
| 16 |
|
| 17 |
+
cmd = f"foundry install rfd3 ligandmpnn rf3 --checkpoint-dir /data/checkpoints"
|
| 18 |
+
print("Global PATH:", os.environ.get("PATH"))
|
| 19 |
+
os.environ["HF_HOME"] = "/data" # HF Spaces persistent dir
|
| 20 |
+
os.environ["FOUNDRY_CHECKPOINT_DIRS"] = "/data/checkpoints"
|
| 21 |
|
| 22 |
result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
|
| 23 |
if result.returncode == 0:
|