chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ui tweak

Browse Source
This commit is contained in:
bennykok 2024-01-28 14:41:14 +08:00
parent 9de266fbab
commit c1fc06fd39
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
web/src/components/custom-form/gpu-picker.tsx
View File

@ -1,5 +1,6 @@
import { AutoFormInputComponentProps } from "@/components/ui/auto-form/types"; import { AutoFormInputComponentProps } from "@/components/ui/auto-form/types";
import { getBaseSchema } from "@/components/ui/auto-form/utils"; import { getBaseSchema } from "@/components/ui/auto-form/utils";
import { Badge } from "@/components/ui/badge";
import { import {
FormItem, FormItem,
FormLabel, FormLabel,
@ -41,6 +42,11 @@ export default function AutoFormGPUPicker({
const plan = fieldConfigItem.inputProps?.sub?.plan; const plan = fieldConfigItem.inputProps?.sub?.plan;
const enabledGPU = ["T4"]; const enabledGPU = ["T4"];
const planMapping: Record<string, string> = {
A10G: "pro",
A100: "enterprise",
};
if (plan == "pro") { if (plan == "pro") {
enabledGPU.push("A10G"); enabledGPU.push("A10G");
} else if (plan == "enterprise") { } else if (plan == "enterprise") {
@ -72,7 +78,9 @@ export default function AutoFormGPUPicker({
{label} {label}
{!enabled && ( {!enabled && (
<span className="mx-2 inline-flex items-center justify-center gap-2"> <span className="mx-2 inline-flex items-center justify-center gap-2">
Upgrade to enabled this options <Lock size={14}></Lock> <Badge className="capitalize">{planMapping[value]}</Badge>{" "}
plan required
<Lock size={14}></Lock>
</span> </span>
)} )}
</SelectItem> </SelectItem>