~ pip fix

model path fix
This commit is contained in:
abc 2024-03-19 12:29:35 +00:00
parent 9a49d06829
commit cc83c6af01

View file

@ -10,7 +10,7 @@ class LocalProvider:
raise ValueError(f"Model '{model}' not found / not yet implemented") raise ValueError(f"Model '{model}' not found / not yet implemented")
model = models[model] model = models[model]
model_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../models/') model_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'models/')
full_model_path = os.path.join(model_dir, model['path']) full_model_path = os.path.join(model_dir, model['path'])
if not os.path.isfile(full_model_path): if not os.path.isfile(full_model_path):
@ -23,7 +23,7 @@ class LocalProvider:
raise ValueError(f"Model '{model['path']}' not found.") raise ValueError(f"Model '{model['path']}' not found.")
model = GPT4All(model_name=model['path'], model = GPT4All(model_name=model['path'],
n_threads=8, #n_threads=8,
verbose=False, verbose=False,
allow_download=False, allow_download=False,
model_path=model_dir) model_path=model_dir)