diff --git a/modules/models.py b/modules/models.py index 0ba584a5..b3e4b8e0 100644 --- a/modules/models.py +++ b/modules/models.py @@ -81,7 +81,7 @@ def load_model(model_name): elif shared.is_RWKV: from modules.RWKV import load_RWKV_model - return load_RWKV_model(Path('models/RWKV-4-Pile-169M-20220807-8023.pth')), None + return load_RWKV_model(Path(f'models/{model_name}')), None # Custom else: