chore: fix responsive and add tensorrt_llm nitro to model utils

LeikoDmitry · Aug 28, 2024 · 31ae14c · 31ae14c
1 parent 46aa9d3
commit 31ae14c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/web/screens/Settings/HuggingFaceRepoDetailModal/ModelDownloadList/index.tsx b/web/screens/Settings/HuggingFaceRepoDetailModal/ModelDownloadList/index.tsx
@@ -29,7 +29,7 @@ const ModelDownloadList = () => {
   return (
     <div className="flex h-[500px] flex-1 flex-col">
       <h1 className="mb-3 font-semibold">Available Versions</h1>
-      <ScrollArea className="h-full w-full lg:flex-1">
+      <ScrollArea className="w-full lg:h-full lg:flex-1">
         {ggufModels.map((model, index) => {
           if (!model.downloadUrl) return null
           return (

diff --git a/web/utils/modelEngine.ts b/web/utils/modelEngine.ts
@@ -4,6 +4,7 @@ export const getLogoEngine = (engine: InferenceEngine) => {
   switch (engine) {
     case InferenceEngine.anthropic:
       return 'images/ModelProvider/anthropic.svg'
+    case InferenceEngine.nitro_tensorrt_llm:
     case InferenceEngine.nitro:
       return 'images/ModelProvider/nitro.svg'
     case InferenceEngine.cortex_llamacpp:
@@ -43,6 +44,8 @@ export const getTitleByEngine = (engine: InferenceEngine) => {
   switch (engine) {
     case InferenceEngine.nitro:
       return 'Llama.cpp (Nitro)'
+    case InferenceEngine.nitro_tensorrt_llm:
+      return 'TensorRT-LLM (Nitro)'
     case InferenceEngine.cortex_llamacpp:
       return 'Llama.cpp (Cortex)'
     case InferenceEngine.cortex_onnx: