~ pip fix

model path fix
This commit is contained in:
abc 2024-03-19 12:29:35 +00:00
parent 9a49d06829
commit cc83c6af01
1 changed files with 2 additions and 2 deletions

View File

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