flxes

transformerlab · Jan 1, 2024 · 017564d · 017564d
1 parent 26eb6b1
commit 017564d
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/src/renderer/components/Experiment/Interact/Interact.tsx b/src/renderer/components/Experiment/Interact/Interact.tsx
@@ -278,7 +278,7 @@ export default function Chat({ experimentInfo, experimentInfoMutate }) {
       undefined,
       undefined,
       undefined,
-      isVLLMInferenceEngine !== null
+      false
     );
     setIsThinking(false);
 

diff --git a/src/renderer/components/Nav/Sidebar.tsx b/src/renderer/components/Nav/Sidebar.tsx
@@ -68,8 +68,7 @@ export default function Sidebar({
     return (
       models?.[0]?.id !==
         experimentInfo?.config?.foundation?.split('/').slice(-1)[0] &&
-      models?.[0]?.id !==
-        `models/peft/${experimentInfo?.config.adaptor}`?.split('/').slice(-1)[0]
+      models?.[0]?.id !== experimentInfo?.config.adaptor
     );
   }
 

diff --git a/src/renderer/lib/transformerlab-api-sdk.ts b/src/renderer/lib/transformerlab-api-sdk.ts
@@ -225,6 +225,8 @@ export async function sendCompletion(
     model = adaptor;
   }
 
+  console.log('model', model);
+
   const data = {
     model: model,
     stream: false, // For streaming responses
@@ -776,7 +778,7 @@ export async function activateWorker(
 
   let model = modelName;
   if (adaptorName !== '') {
-    model = `workspace/peft/${modelName}/${adaptorName}`;
+    model = `workspace/adaptors/${modelName}/${adaptorName}`;
   }
 
   if (modelFilename !== null) {