From b870b4a56533ae74e1c8aeee82728b0c8518afc8 Mon Sep 17 00:00:00 2001
From: sanjaycal <sanjay.calgary@gmail.com>
Date: Thu, 27 Feb 2025 17:33:42 -0500
Subject: [PATCH] add downloads to frontend
---
.../Experiment/Workflows/NewNodeModal.tsx | 22 +++++++++++++++++++
1 file changed, 22 insertions(+)
diff --git a/src/renderer/components/Experiment/Workflows/NewNodeModal.tsx b/src/renderer/components/Experiment/Workflows/NewNodeModal.tsx
index f12c6316..47cb4698 100644
--- a/src/renderer/components/Experiment/Workflows/NewNodeModal.tsx
+++ b/src/renderer/components/Experiment/Workflows/NewNodeModal.tsx
@@ -85,6 +85,21 @@ export default function NewNodeModal({
JSON.stringify(node)
)
);
+ } else if (mode == 'DOWNLOAD_MODEL') {
+ const model = formData.get('model') as string;
+ const config = JSON.parse(selectedWorkflow.config);
+ console.log(config);
+ const node = {
+ name: name,
+ type: 'DOWNLOAD_MODEL',
+ model: model,
+ };
+ await fetch(
+ chatAPI.Endpoints.Workflows.AddNode(
+ selectedWorkflow.id,
+ JSON.stringify(node)
+ )
+ );
} else {
const node = JSON.parse(formData.get('node') as string);
node.name = name;
@@ -112,6 +127,7 @@ export default function NewNodeModal({
onChange={handleModeChange}
required
>
+ <Option value="DOWNLOAD_MODEL">DOWNLOAD MODEL</Option>
<Option value="TRAIN">TRAIN</Option>
<Option value="EVAL">EVAL</Option>
<Option value="OTHER">OTHER</Option>
@@ -119,6 +135,12 @@ export default function NewNodeModal({
</FormControl>
<FormControl>
+ {mode == 'DOWNLOAD_MODEL' && (
+ <>
+ <FormLabel>Huggingface Model Id</FormLabel>
+ <Input autoFocus required name="model" />
+ </>
+ )}
{mode == 'TRAIN' && (
<>
<FormLabel>Training Template</FormLabel>
--
GitLab