From 6837d4d72a759c88696db87f11e1b6ae82a6db6b Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Tue, 28 Feb 2023 02:52:29 -0300
Subject: [PATCH] Load the model by name

---
 modules/models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: