added revision

Signed-off-by: Matteo Omenetti <[email protected]>
DS4SD · Jan 21, 2025 · 27e5aa9 · 27e5aa9
1 parent a9be473
commit 27e5aa9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_code_formula_predictor.py b/tests/test_code_formula_predictor.py
@@ -37,8 +37,7 @@ def init() -> dict:
     }
 
     # Download models from HF
-    download_path = snapshot_download(repo_id="ds4sd/CodeFormula")
-    artifact_path = os.path.join(download_path, "")
+    artifact_path = snapshot_download(repo_id="ds4sd/CodeFormula", revision="v1.0.0")
 
     init["artifact_path"] = artifact_path