Spaces:
Sleeping
Sleeping
ehristoforu
commited on
Commit
•
2d56a10
1
Parent(s):
ea5f41e
Update modules/load_model.py
Browse files- modules/load_model.py +1 -1
modules/load_model.py
CHANGED
@@ -12,7 +12,7 @@ def load_model(path, n_ctx, n_gpu_layers, n_threads, verbose, f16_kv, logits_all
|
|
12 |
dir = os.getcwd()
|
13 |
global llm
|
14 |
llm = Llama(
|
15 |
-
model_path=f"{dir}
|
16 |
n_ctx=n_ctx,
|
17 |
n_gpu_layers=n_gpu_layers,
|
18 |
n_threads=n_threads,
|
|
|
12 |
dir = os.getcwd()
|
13 |
global llm
|
14 |
llm = Llama(
|
15 |
+
model_path=f"{dir}/models/{path}",
|
16 |
n_ctx=n_ctx,
|
17 |
n_gpu_layers=n_gpu_layers,
|
18 |
n_threads=n_threads,
|