fix

openclimatefix · Jul 26, 2024 · 067283a · 067283a
1 parent f4c1c93
commit 067283a
Showing 1 changed file with 1 addition and 1 deletion.
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)