diff --git a/modeling/inference_models/rwkv.py b/modeling/inference_models/rwkv.py index ea290841..e1f984e2 100644 --- a/modeling/inference_models/rwkv.py +++ b/modeling/inference_models/rwkv.py @@ -139,7 +139,7 @@ class RWKVInferenceModel(InferenceModel): # Download model if we need to model_path = os.path.join(model_dir, MODEL_FILES[self.model_name]) if not os.path.exists(model_path): - self.download_model(self, model_path, model_class) + self._download_model(self, model_path, model_class) # Now we load!