Timothy J. Baek 11 months ago
parent
commit
a1888b3757
2 changed files with 2 additions and 1 deletions
  1. 1 1
      backend/apps/ollama/main.py
  2. 1 0
      backend/utils/misc.py

+ 1 - 1
backend/apps/ollama/main.py

@@ -43,8 +43,8 @@ from config import (
     AppConfig,
 )
 from utils.misc import (
-    apply_model_params_to_body_ollama,
     calculate_sha256,
+    apply_model_params_to_body_ollama,
     apply_model_params_to_body_openai,
     apply_model_system_prompt_to_body,
 )

+ 1 - 0
backend/utils/misc.py

@@ -179,6 +179,7 @@ def apply_model_params_to_body_ollama(params: dict, form_data: dict) -> dict:
         "use_mmap",
         "use_mlock",
         "num_thread",
+        "num_gpu",
     ]
     mappings = {i: lambda x: x for i in opts}
     form_data = apply_model_params_to_body(params, form_data, mappings)