Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: fix responsive and add tensorrt_llm nitro to model utils

Browse Source
This commit is contained in:
Faisal Amir 2024-08-28 17:04:47 +07:00
parent 46aa9d3a29
commit 31ae14c1f1
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
web/screens/Settings/HuggingFaceRepoDetailModal/ModelDownloadList/index.tsx
View File

@ -29,7 +29,7 @@ const ModelDownloadList = () => {
return ( return (
<div className="flex h-[500px] flex-1 flex-col"> <div className="flex h-[500px] flex-1 flex-col">
<h1 className="mb-3 font-semibold">Available Versions</h1> <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) => { {ggufModels.map((model, index) => {
if (!model.downloadUrl) return null if (!model.downloadUrl) return null
return ( return (

3
web/utils/modelEngine.ts
View File

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