Commit a1888b37 authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

refac

parent c75e77cd
...@@ -43,8 +43,8 @@ from config import ( ...@@ -43,8 +43,8 @@ from config import (
AppConfig, AppConfig,
) )
from utils.misc import ( from utils.misc import (
apply_model_params_to_body_ollama,
calculate_sha256, calculate_sha256,
apply_model_params_to_body_ollama,
apply_model_params_to_body_openai, apply_model_params_to_body_openai,
apply_model_system_prompt_to_body, apply_model_system_prompt_to_body,
) )
......
...@@ -179,6 +179,7 @@ def apply_model_params_to_body_ollama(params: dict, form_data: dict) -> dict: ...@@ -179,6 +179,7 @@ def apply_model_params_to_body_ollama(params: dict, form_data: dict) -> dict:
"use_mmap", "use_mmap",
"use_mlock", "use_mlock",
"num_thread", "num_thread",
"num_gpu",
] ]
mappings = {i: lambda x: x for i in opts} mappings = {i: lambda x: x for i in opts}
form_data = apply_model_params_to_body(params, form_data, mappings) form_data = apply_model_params_to_body(params, form_data, mappings)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment