Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 18:45:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Load the model by name

Browse source
This commit is contained in:
oobabooga 2023-02-28 02:52:29 -03:00
parent f871971de1
commit 6837d4d72a
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/models.py
View file

@ -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: