Merge pull request #24 from openclimatefix/update-hf

update pvnet model in tests
openclimatefix · Jul 26, 2024 · f66d3fc · f66d3fc
2 parents 99defdc + 2d3ba9a
commit f66d3fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pvnet_summation/models/base_model.py b/pvnet_summation/models/base_model.py
@@ -50,7 +50,7 @@ def __init__(
         self.pvnet_model_version = model_version
 
         self.pvnet_model = PVNetBaseModel.from_pretrained(
-            model_name,
+            model_id=model_name,
             revision=model_version,
         )
         self.pvnet_model.requires_grad_(False)

diff --git a/pyproject.toml b/pyproject.toml
@@ -27,7 +27,7 @@ dependencies = [
     "omegaconf",
     "hydra-core",
     "python-dotenv",
-    "huggingface-hub==0.20.*",
+    "huggingface-hub",
     "geopandas==0.14.4",
 ]