AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix path on models.py

Browse Source
This commit is contained in:
oobabooga 2023-04-26 03:29:09 -03:00
parent a777c058af
commit 9c2e7c0fab
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

@ -56,7 +56,7 @@ def find_model_type(model_name):
elif any((k in model_name_lower for k in ['gpt4chan', 'gpt-4chan'])): elif any((k in model_name_lower for k in ['gpt4chan', 'gpt-4chan'])):
return 'gpt4chan' return 'gpt4chan'
else: else:
config = AutoConfig.from_pretrained(f"{shared.args.model_dir}/{model_name}") config = AutoConfig.from_pretrained(Path(f'{shared.args.model_dir}/{model_name}'))
# Not a "catch all", but fairly accurate # Not a "catch all", but fairly accurate
if config.to_dict().get("is_encoder_decoder", False): if config.to_dict().get("is_encoder_decoder", False):
return 'HF_seq2seq' return 'HF_seq2seq'