Hansimov commited on
Commit
584ffa5
1 Parent(s): 6828f2e

:zap: [Enhance] Prettify model name for local llama-cpp llm

Browse files
Files changed (1) hide show
  1. storages/endpoint_storage.js +1 -0
storages/endpoint_storage.js CHANGED
@@ -225,6 +225,7 @@ class EndpointStorage {
225
  let endpoint_hostname = new URL(endpoint).hostname
226
  .split(".")[0]
227
  .split("-")[0];
 
228
  let model_name = `${model_id} (${endpoint_hostname})`;
229
  let model_value = `${endpoint}|${model_id}`;
230
  return [model_name, model_value];
 
225
  let endpoint_hostname = new URL(endpoint).hostname
226
  .split(".")[0]
227
  .split("-")[0];
228
+ model_id = model_id.split("/").pop().split(".")[0];
229
  let model_name = `${model_id} (${endpoint_hostname})`;
230
  let model_value = `${endpoint}|${model_id}`;
231
  return [model_name, model_value];