diff --git a/ui/src/app/jobs/new/AdvancedJob.tsx b/ui/src/app/jobs/new/AdvancedJob.tsx
new file mode 100644
index 00000000..9d0873e1
--- /dev/null
+++ b/ui/src/app/jobs/new/AdvancedJob.tsx
@@ -0,0 +1,148 @@
+'use client';
+import { useEffect, useState, useRef } from 'react';
+import { JobConfig } from '@/types';
+import YAML from 'yaml';
+import Editor, { OnMount } from '@monaco-editor/react';
+import type { editor } from 'monaco-editor';
+import { Settings } from '@/hooks/useSettings';
+
+type Props = {
+  jobConfig: JobConfig;
+  setJobConfig: (value: any, key?: string) => void;
+  status: 'idle' | 'saving' | 'success' | 'error';
+  handleSubmit: (event: React.FormEvent<HTMLFormElement>) => void;
+  runId: string | null;
+  gpuIDs: string | null;
+  setGpuIDs: (value: string | null) => void;
+  gpuList: any;
+  datasetOptions: any;
+  settings: Settings;
+};
+
+const isDev = process.env.NODE_ENV === 'development';
+
+const yamlConfig: YAML.DocumentOptions &
+  YAML.SchemaOptions &
+  YAML.ParseOptions &
+  YAML.CreateNodeOptions &
+  YAML.ToStringOptions = {
+  indent: 2,
+  lineWidth: 999999999999,
+  defaultStringType: 'QUOTE_DOUBLE',
+  defaultKeyType: 'PLAIN',
+  directives: true,
+};
+
+export default function AdvancedJob({
+  jobConfig,
+  setJobConfig,
+  settings,
+}: Props) {
+  const [editorValue, setEditorValue] = useState<string>('');
+  const lastJobConfigUpdateStringRef = useRef('');
+  const editorRef = useRef<editor.IStandaloneCodeEditor | null>(null);
+
+  // Track if the editor has been mounted
+  const isEditorMounted = useRef(false);
+
+  // Handler for editor mounting
+  const handleEditorDidMount: OnMount = editor => {
+    editorRef.current = editor;
+    isEditorMounted.current = true;
+
+    // Initial content setup
+    try {
+      const yamlContent = YAML.stringify(jobConfig, yamlConfig);
+      setEditorValue(yamlContent);
+      lastJobConfigUpdateStringRef.current = JSON.stringify(jobConfig);
+    } catch (e) {
+      console.warn(e);
+    }
+  };
+
+  useEffect(() => {
+    const lastUpdate = lastJobConfigUpdateStringRef.current;
+    const currentUpdate = JSON.stringify(jobConfig);
+
+    // Skip if no changes or editor not yet mounted
+    if (lastUpdate === currentUpdate || !isEditorMounted.current) {
+      return;
+    }
+
+    try {
+      // Preserve cursor position and selection
+      const editor = editorRef.current;
+      if (editor) {
+        // Save current editor state
+        const position = editor.getPosition();
+        const selection = editor.getSelection();
+        const scrollTop = editor.getScrollTop();
+
+        // Update content
+        const yamlContent = YAML.stringify(jobConfig, yamlConfig);
+
+        // Only update if the content is actually different
+        if (yamlContent !== editor.getValue()) {
+          // Set value directly on the editor model instead of using React state
+          editor.getModel()?.setValue(yamlContent);
+
+          // Restore cursor position and selection
+          if (position) editor.setPosition(position);
+          if (selection) editor.setSelection(selection);
+          editor.setScrollTop(scrollTop);
+        }
+
+        lastJobConfigUpdateStringRef.current = currentUpdate;
+      }
+    } catch (e) {
+      console.warn(e);
+    }
+  }, [jobConfig]);
+
+  const handleChange = (value: string | undefined) => {
+    if (value === undefined) return;
+
+    try {
+      const parsed = YAML.parse(value);
+      // Don't update jobConfig if the change came from the editor itself
+      // to avoid a circular update loop
+      if (JSON.stringify(parsed) !== lastJobConfigUpdateStringRef.current) {
+        lastJobConfigUpdateStringRef.current = JSON.stringify(parsed);
+
+        // We have to ensure certain things are always set
+        try {
+          parsed.config.process[0].type = 'ui_trainer';
+          parsed.config.process[0].sqlite_db_path = './aitk_db.db';
+          parsed.config.process[0].training_folder = settings.TRAINING_FOLDER;
+          parsed.config.process[0].device = 'cuda';
+          parsed.config.process[0].performance_log_every = 10;
+        } catch (e) {
+          console.warn(e);
+        }
+        setJobConfig(parsed);
+      }
+    } catch (e) {
+      // Don't update on parsing errors
+      console.warn(e);
+    }
+  };
+
+  return (
+    <>
+      <Editor
+        height="100%"
+        width="100%"
+        defaultLanguage="yaml"
+        value={editorValue}
+        theme="vs-dark"
+        onChange={handleChange}
+        onMount={handleEditorDidMount}
+        options={{
+          minimap: { enabled: true },
+          scrollBeyondLastLine: false,
+          automaticLayout: true,
+        }}
+      />
+    </>
+  );
+}
diff --git a/ui/src/app/jobs/new/SimpleJob.tsx b/ui/src/app/jobs/new/SimpleJob.tsx
new file mode 100644
index 00000000..75bab27d
--- /dev/null
+++ b/ui/src/app/jobs/new/SimpleJob.tsx
@@ -0,0 +1,627 @@
+'use client';
+
+import { options, modelArchs, isVideoModelFromArch } from './options';
+import { defaultDatasetConfig } from './jobConfig';
+import { JobConfig } from '@/types';
+import { objectCopy } from '@/utils/basic';
+import { TextInput, SelectInput, Checkbox, FormGroup, NumberInput } from '@/components/formInputs';
+import Card from '@/components/Card';
+import { X } from 'lucide-react';
+
+type Props = {
+  jobConfig: JobConfig;
+  setJobConfig: (value: any, key: string) => void;
+  status: 'idle' | 'saving' | 'success' | 'error';
+  handleSubmit: (event: React.FormEvent<HTMLFormElement>) => void;
+  runId: string | null;
+  gpuIDs: string | null;
+  setGpuIDs: (value: string | null) => void;
+  gpuList: any;
+  datasetOptions: any;
+};
+
+const isDev = process.env.NODE_ENV === 'development';
+
+export default function SimpleJob({
+  jobConfig,
+  setJobConfig,
+  handleSubmit,
+  status,
+  runId,
+  gpuIDs,
+  setGpuIDs,
+  gpuList,
+  datasetOptions,
+}: Props) {
+  const isVideoModel = isVideoModelFromArch(jobConfig.config.process[0].model.arch);
+  return (
+    <>
+      <form onSubmit={handleSubmit} className="space-y-8">
+        <div className="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-4 gap-6">
+          <Card title="Job Settings">
+            <TextInput
+              label="Training Name"
+              value={jobConfig.config.name}
+              onChange={value => setJobConfig(value, 'config.name')}
+              placeholder="Enter training name"
+              disabled={runId !== null}
+              required
+            />
+            <SelectInput
+              label="GPU ID"
+              value={`${gpuIDs}`}
+              onChange={value => setGpuIDs(value)}
+              options={gpuList.map((gpu: any) => ({ value: `${gpu.index}`, label: `GPU #${gpu.index}` }))}
+            />
+            <TextInput
+              label="Trigger Word"
+              value={jobConfig.config.process[0].trigger_word || ''}
+              onChange={(value: string | null) => {
+                if (value?.trim() === '') {
+                  value = null;
+                }
+                setJobConfig(value, 'config.process[0].trigger_word');
+              }}
+              placeholder=""
+              required
+            />
+          </Card>
+
+          {/* Model Configuration Section */}
+          <Card title="Model Configuration">
+            <SelectInput
+              label="Name or Path"
+              value={jobConfig.config.process[0].model.name_or_path}
+              onChange={value => {
+                // see if model changed
+                const currentModel = options.model.find(
+                  model => model.name_or_path === jobConfig.config.process[0].model.name_or_path,
+                );
+                if (!currentModel || currentModel.name_or_path === value) {
+                  // model has not changed
+                  return;
+                }
+                // revert defaults from previous model
+                for (const key in currentModel.defaults) {
+                  setJobConfig(currentModel.defaults[key][1], key);
+                }
+                // set new model
+                setJobConfig(value, 'config.process[0].model.name_or_path');
+                // update the defaults when a model is selected
+                const model = options.model.find(model => model.name_or_path === value);
+                if (model?.defaults) {
+                  for (const key in model.defaults) {
+                    setJobConfig(model.defaults[key][0], key);
+                  }
+                }
+              }}
+              options={
+                options.model
+                  .map(model => {
+                    if (model.dev_only && !isDev) {
+                      return null;
+                    }
+                    return {
+                      value: model.name_or_path,
+                      label: model.name_or_path,
+                    };
+                  })
+                  .filter(x => x) as { value: string; label: string }[]
+              }
+            />
+            <SelectInput
+              label="Model Architecture"
+              value={jobConfig.config.process[0].model.arch}
+              onChange={value => {
+                const currentArch = modelArchs.find(a => a.name === jobConfig.config.process[0].model.arch);
+                if (!currentArch || currentArch.name === value) {
+                  return;
+                }
+                // set new model
+                setJobConfig(value, 'config.process[0].model.arch');
+              }}
+              options={
+                modelArchs
+                  .map(model => {
+                    return {
+                      value: model.name,
+                      label: model.label,
+                    };
+                  })
+                  .filter(x => x) as { value: string; label: string }[]
+              }
+            />
+            <FormGroup label="Quantize">
+              <div className="grid grid-cols-2 gap-2">
+                <Checkbox
+                  label="Transformer"
+                  checked={jobConfig.config.process[0].model.quantize}
+                  onChange={value => setJobConfig(value, 'config.process[0].model.quantize')}
+                />
+                <Checkbox
+                  label="Text Encoder"
+                  checked={jobConfig.config.process[0].model.quantize_te}
+                  onChange={value => setJobConfig(value, 'config.process[0].model.quantize_te')}
+                />
+              </div>
+            </FormGroup>
+          </Card>
+          <Card title="Target Configuration">
+            <SelectInput
+              label="Target Type"
+              value={jobConfig.config.process[0].network?.type ?? 'lora'}
+              onChange={value => setJobConfig(value, 'config.process[0].network.type')}
+              options={[
+                { value: 'lora', label: 'LoRA' },
+                { value: 'lokr', label: 'LoKr' },
+              ]}
+            />
+            {jobConfig.config.process[0].network?.type == 'lokr' && (
+              <SelectInput
+                label="LoKr Factor"
+                value={`${jobConfig.config.process[0].network?.lokr_factor ?? -1}`}
+                onChange={value => setJobConfig(parseInt(value), 'config.process[0].network.lokr_factor')}
+                options={[
+                  { value: '-1', label: 'Auto' },
+                  { value: '4', label: '4' },
+                  { value: '8', label: '8' },
+                  { value: '16', label: '16' },
+                  { value: '32', label: '32' },
+                ]}
+              />
+            )}
+            {jobConfig.config.process[0].network?.type == 'lora' && (
+              <NumberInput
+                label="Linear Rank"
+                value={jobConfig.config.process[0].network.linear}
+                onChange={value => {
+                  console.log('onChange', value);
+                  setJobConfig(value, 'config.process[0].network.linear');
+                  setJobConfig(value, 'config.process[0].network.linear_alpha');
+                }}
+                placeholder="eg. 16"
+                min={0}
+                max={1024}
+                required
+              />
+            )}
+          </Card>
+          <Card title="Save Configuration">
+            <SelectInput
+              label="Data Type"
+              value={jobConfig.config.process[0].save.dtype}
+              onChange={value => setJobConfig(value, 'config.process[0].save.dtype')}
+              options={[
+                { value: 'bf16', label: 'BF16' },
+                { value: 'fp16', label: 'FP16' },
+                { value: 'fp32', label: 'FP32' },
+              ]}
+            />
+            <NumberInput
+              label="Save Every"
+              value={jobConfig.config.process[0].save.save_every}
+              onChange={value => setJobConfig(value, 'config.process[0].save.save_every')}
+              placeholder="eg. 250"
+              min={1}
+              required
+            />
+            <NumberInput
+              label="Max Step Saves to Keep"
+              value={jobConfig.config.process[0].save.max_step_saves_to_keep}
+              onChange={value => setJobConfig(value, 'config.process[0].save.max_step_saves_to_keep')}
+              placeholder="eg. 4"
+              min={1}
+              required
+            />
+          </Card>
+        </div>
+        <div>
+          <Card title="Training Configuration">
+            <div className="grid grid-cols-1 md:grid-cols-3 lg:grid-cols-5 gap-6">
+              <div>
+                <NumberInput
+                  label="Batch Size"
+                  value={jobConfig.config.process[0].train.batch_size}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.batch_size')}
+                  placeholder="eg. 4"
+                  min={1}
+                  required
+                />
+                <NumberInput
+                  label="Gradient Accumulation"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.gradient_accumulation}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.gradient_accumulation')}
+                  placeholder="eg. 1"
+                  min={1}
+                  required
+                />
+                <NumberInput
+                  label="Steps"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.steps}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.steps')}
+                  placeholder="eg. 2000"
+                  min={1}
+                  required
+                />
+              </div>
+              <div>
+                <SelectInput
+                  label="Optimizer"
+                  value={jobConfig.config.process[0].train.optimizer}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.optimizer')}
+                  options={[
+                    { value: 'adamw8bit', label: 'AdamW8Bit' },
+                    { value: 'adafactor', label: 'Adafactor' },
+                  ]}
+                />
+                <NumberInput
+                  label="Learning Rate"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.lr}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.lr')}
+                  placeholder="eg. 0.0001"
+                  min={0}
+                  required
+                />
+                <NumberInput
+                  label="Weight Decay"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.optimizer_params.weight_decay}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.optimizer_params.weight_decay')}
+                  placeholder="eg. 0.0001"
+                  min={0}
+                  required
+                />
+              </div>
+              <div>
+                <SelectInput
+                  label="Timestep Type"
+                  value={jobConfig.config.process[0].train.timestep_type}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.timestep_type')}
+                  options={[
+                    { value: 'sigmoid', label: 'Sigmoid' },
+                    { value: 'linear', label: 'Linear' },
+                    { value: 'flux_shift', label: 'Flux Shift' },
+                  ]}
+                />
+                <SelectInput
+                  label="Timestep Bias"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.content_or_style}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.content_or_style')}
+                  options={[
+                    { value: 'balanced', label: 'Balanced' },
+                    { value: 'content', label: 'High Noise' },
+                    { value: 'style', label: 'Low Noise' },
+                  ]}
+                />
+                <SelectInput
+                  label="Noise Scheduler"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.noise_scheduler}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.noise_scheduler')}
+                  options={[
+                    { value: 'flowmatch', label: 'FlowMatch' },
+                    { value: 'ddpm', label: 'DDPM' },
+                  ]}
+                />
+              </div>
+              <div>
+                <FormGroup label="EMA (Exponential Moving Average)">
+                  <Checkbox
+                    label="Use EMA"
+                    className="pt-1"
+                    checked={jobConfig.config.process[0].train.ema_config?.use_ema || false}
+                    onChange={value => setJobConfig(value, 'config.process[0].train.ema_config.use_ema')}
+                  />
+                </FormGroup>
+                <NumberInput
+                  label="EMA Decay"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.ema_config?.ema_decay as number}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.ema_config?.ema_decay')}
+                  placeholder="eg. 0.99"
+                  min={0}
+                />
+                <FormGroup label="Unload Text Encoder" className="pt-2">
+                  <div className="grid grid-cols-2 gap-2">
+                    <Checkbox
+                      label="Unload TE"
+                      checked={jobConfig.config.process[0].train.unload_text_encoder || false}
+                      onChange={value => setJobConfig(value, 'config.process[0].train.unload_text_encoder')}
+                    />
+                  </div>
+                </FormGroup>
+              </div>
+              <div>
+                <FormGroup label="Regularization">
+                  <Checkbox
+                    label="Differtial Output Preservation"
+                    className="pt-1"
+                    checked={jobConfig.config.process[0].train.diff_output_preservation || false}
+                    onChange={value => setJobConfig(value, 'config.process[0].train.diff_output_preservation')}
+                  />
+                </FormGroup>
+                <NumberInput
+                  label="DFE Loss Multiplier"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.diff_output_preservation_multiplier as number}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.diff_output_preservation_multiplier')}
+                  placeholder="eg. 1.0"
+                  min={0}
+                />
+                <TextInput
+                  label="DFE Preservation Class"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].train.diff_output_preservation_class as string}
+                  onChange={value => setJobConfig(value, 'config.process[0].train.diff_output_preservation_class')}
+                  placeholder="eg. woman"
+                />
+              </div>
+            </div>
+          </Card>
+        </div>
+        <div>
+          <Card title="Datasets">
+            <>
+              {jobConfig.config.process[0].datasets.map((dataset, i) => (
+                <div key={i} className="p-4 rounded-lg bg-gray-800 relative">
+                  <button
+                    type="button"
+                    onClick={() =>
+                      setJobConfig(
+                        jobConfig.config.process[0].datasets.filter((_, index) => index !== i),
+                        'config.process[0].datasets',
+                      )
+                    }
+                    className="absolute top-2 right-2 bg-red-800 hover:bg-red-700 rounded-full p-1 text-sm transition-colors"
+                  >
+                    <X />
+                  </button>
+                  <h2 className="text-lg font-bold mb-4">Dataset {i + 1}</h2>
+                  <div className="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-4 gap-6">
+                    <div>
+                      <SelectInput
+                        label="Dataset"
+                        value={dataset.folder_path}
+                        onChange={value => setJobConfig(value, `config.process[0].datasets[${i}].folder_path`)}
+                        options={datasetOptions}
+                      />
+                      <NumberInput
+                        label="LoRA Weight"
+                        value={dataset.network_weight}
+                        className="pt-2"
+                        onChange={value => setJobConfig(value, `config.process[0].datasets[${i}].network_weight`)}
+                        placeholder="eg. 1.0"
+                      />
+                    </div>
+                    <div>
+                      <TextInput
+                        label="Default Caption"
+                        value={dataset.default_caption}
+                        onChange={value => setJobConfig(value, `config.process[0].datasets[${i}].default_caption`)}
+                        placeholder="eg. A photo of a cat"
+                      />
+                      <NumberInput
+                        label="Caption Dropout Rate"
+                        className="pt-2"
+                        value={dataset.caption_dropout_rate}
+                        onChange={value => setJobConfig(value, `config.process[0].datasets[${i}].caption_dropout_rate`)}
+                        placeholder="eg. 0.05"
+                        min={0}
+                        required
+                      />
+                    </div>
+                    <div>
+                      <FormGroup label="Settings" className="">
+                        <Checkbox
+                          label="Cache Latents"
+                          checked={dataset.cache_latents_to_disk || false}
+                          onChange={value =>
+                            setJobConfig(value, `config.process[0].datasets[${i}].cache_latents_to_disk`)
+                          }
+                        />
+                        <Checkbox
+                          label="Is Regularization"
+                          checked={dataset.is_reg || false}
+                          onChange={value => setJobConfig(value, `config.process[0].datasets[${i}].is_reg`)}
+                        />
+                      </FormGroup>
+                    </div>
+                    <div>
+                      <FormGroup label="Resolutions" className="pt-2">
+                        <div className="grid grid-cols-2 gap-2">
+                          {[
+                            [256, 512, 768],
+                            [1024, 1280, 1536],
+                          ].map(resGroup => (
+                            <div key={resGroup[0]} className="space-y-2">
+                              {resGroup.map(res => (
+                                <Checkbox
+                                  key={res}
+                                  label={res.toString()}
+                                  checked={dataset.resolution.includes(res)}
+                                  onChange={value => {
+                                    const resolutions = dataset.resolution.includes(res)
+                                      ? dataset.resolution.filter(r => r !== res)
+                                      : [...dataset.resolution, res];
+                                    setJobConfig(resolutions, `config.process[0].datasets[${i}].resolution`);
+                                  }}
+                                />
+                              ))}
+                            </div>
+                          ))}
+                        </div>
+                      </FormGroup>
+                    </div>
+                  </div>
+                </div>
+              ))}
+              <button
+                type="button"
+                onClick={() =>
+                  setJobConfig(
+                    [...jobConfig.config.process[0].datasets, objectCopy(defaultDatasetConfig)],
+                    'config.process[0].datasets',
+                  )
+                }
+                className="w-full px-4 py-2 bg-gray-700 hover:bg-gray-600 rounded-lg transition-colors"
+              >
+                Add Dataset
+              </button>
+            </>
+          </Card>
+        </div>
+        <div>
+          <Card title="Sample Configuration">
+            <div
+              className={
+                isVideoModel
+                  ? 'grid grid-cols-1 md:grid-cols-3 lg:grid-cols-5 gap-6'
+                  : 'grid grid-cols-1 md:grid-cols-2 lg:grid-cols-4 gap-6'
+              }
+            >
+              <div>
+                <NumberInput
+                  label="Sample Every"
+                  value={jobConfig.config.process[0].sample.sample_every}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.sample_every')}
+                  placeholder="eg. 250"
+                  min={1}
+                  required
+                />
+                <SelectInput
+                  label="Sampler"
+                  className="pt-2"
+                  value={jobConfig.config.process[0].sample.sampler}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.sampler')}
+                  options={[
+                    { value: 'flowmatch', label: 'FlowMatch' },
+                    { value: 'ddpm', label: 'DDPM' },
+                  ]}
+                />
+              </div>
+              <div>
+                <NumberInput
+                  label="Guidance Scale"
+                  value={jobConfig.config.process[0].sample.guidance_scale}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.guidance_scale')}
+                  placeholder="eg. 1.0"
+                  min={0}
+                  required
+                />
+                <NumberInput
+                  label="Sample Steps"
+                  value={jobConfig.config.process[0].sample.sample_steps}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.sample_steps')}
+                  placeholder="eg. 1"
+                  className="pt-2"
+                  min={1}
+                  required
+                />
+              </div>
+              <div>
+                <NumberInput
+                  label="Width"
+                  value={jobConfig.config.process[0].sample.width}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.width')}
+                  placeholder="eg. 1024"
+                  min={0}
+                  required
+                />
+                <NumberInput
+                  label="Height"
+                  value={jobConfig.config.process[0].sample.height}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.height')}
+                  placeholder="eg. 1024"
+                  className="pt-2"
+                  min={0}
+                  required
+                />
+              </div>
+
+              <div>
+                <NumberInput
+                  label="Seed"
+                  value={jobConfig.config.process[0].sample.seed}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.seed')}
+                  placeholder="eg. 0"
+                  min={0}
+                  required
+                />
+                <Checkbox
+                  label="Walk Seed"
+                  className="pt-4 pl-2"
+                  checked={jobConfig.config.process[0].sample.walk_seed}
+                  onChange={value => setJobConfig(value, 'config.process[0].sample.walk_seed')}
+                />
+              </div>
+              {isVideoModel && (
+                <div>
+                  <NumberInput
+                    label="Num Frames"
+                    value={jobConfig.config.process[0].sample.num_frames}
+                    onChange={value => setJobConfig(value, 'config.process[0].sample.num_frames')}
+                    placeholder="eg. 0"
+                    min={0}
+                    required
+                  />
+                  <NumberInput
+                    label="FPS"
+                    value={jobConfig.config.process[0].sample.fps}
+                    onChange={value => setJobConfig(value, 'config.process[0].sample.fps')}
+                    placeholder="eg. 0"
+                    min={0}
+                    required
+                  />
+                </div>
+              )}
+            </div>
+            <FormGroup label={`Sample Prompts (${jobConfig.config.process[0].sample.prompts.length})`} className="pt-2">
+              {jobConfig.config.process[0].sample.prompts.map((prompt, i) => (
+                <div key={i} className="flex items-center space-x-2">
+                  <div className="flex-1">
+                    <TextInput
+                      value={prompt}
+                      onChange={value => setJobConfig(value, `config.process[0].sample.prompts[${i}]`)}
+                      placeholder="Enter prompt"
+                      required
+                    />
+                  </div>
+                  <div>
+                    <button
+                      type="button"
+                      onClick={() =>
+                        setJobConfig(
+                          jobConfig.config.process[0].sample.prompts.filter((_, index) => index !== i),
+                          'config.process[0].sample.prompts',
+                        )
+                      }
+                      className="rounded-full p-1 text-sm"
+                    >
+                      <X />
+                    </button>
+                  </div>
+                </div>
+              ))}
+              <button
+                type="button"
+                onClick={() =>
+                  setJobConfig([...jobConfig.config.process[0].sample.prompts, ''], 'config.process[0].sample.prompts')
+                }
+                className="w-full px-4 py-2 bg-gray-700 hover:bg-gray-600 rounded-lg transition-colors"
+              >
+                Add Prompt
+              </button>
+            </FormGroup>
+          </Card>
+        </div>
+
+        {status === 'success' && <p className="text-green-500 text-center">Training saved successfully!</p>}
+        {status === 'error' && <p className="text-red-500 text-center">Error saving training. Please try again.</p>}
+      </form>
+    </>
+  );
+}