diff --git a/ui/src/app/jobs/new/SimpleJob.tsx b/ui/src/app/jobs/new/SimpleJob.tsx index 9d192f67..dcf38314 100644 --- a/ui/src/app/jobs/new/SimpleJob.tsx +++ b/ui/src/app/jobs/new/SimpleJob.tsx @@ -1,6 +1,6 @@ 'use client'; import { useMemo } from 'react'; -import { modelArchs, ModelArch, groupedModelOptions } from './options'; +import { modelArchs, ModelArch, groupedModelOptions, quantizationOptions, defaultQtype } from './options'; import { defaultDatasetConfig } from './jobConfig'; import { JobConfig } from '@/types'; import { objectCopy } from '@/utils/basic'; @@ -40,10 +40,16 @@ export default function SimpleJob({ const isVideoModel = !!(modelArch?.group === 'video'); + let topBarClass = 'grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 xl:grid-cols-5 gap-6'; + + if (modelArch?.disableSections?.includes('model.quantize')) { + topBarClass = 'grid grid-cols-1 md:grid-cols-2 lg:grid-cols-4 xl:grid-cols-4 gap-6'; + } + return ( <>