taxabind update

mvrl · Oct 29, 2024 · 7aeb0c7 · 7aeb0c7
1 parent 449c6e1
commit 7aeb0c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rshf/taxabind/model.py b/rshf/taxabind/model.py
@@ -38,10 +38,10 @@ def get_audio_processor(self):
         return self.process_audio
 
     def get_location_encoder(self):
-        return GeoCLIP(self.config.location_encoder)
+        return GeoCLIP.from_pretrained(self.config.location_encoder)
 
     def get_env_encoder(self):
-        return SINR(self.config.sinr)
+        return SINR.from_pretrained(self.config.sinr)
 
     def get_env_processor(self):
         return SINR.preprocess_locs