github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add compile to the actual right config section.

Browse Source
This commit is contained in:
Jaret Burkett
2026-06-12 11:12:41 -06:00
parent c09b228a35
commit a5558ae7d9
3 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ui/src/app/jobs/new/SimpleJob.tsx
View File

@@ -406,16 +406,16 @@ export default function SimpleJob({
</FormGroup> </FormGroup>
<Checkbox <Checkbox
label="Compile Model" label="Compile Model"
checked={jobConfig.config.process[0].train.compile || false} checked={jobConfig.config.process[0].model.compile || false}
onChange={value => { onChange={value => {
setJobConfig(value, 'config.process[0].train.compile'); setJobConfig(value, 'config.process[0].model.compile');
if (value) { if (value) {
for (const key in defaultCompileOptions) { for (const key in defaultCompileOptions) {
setJobConfig((defaultCompileOptions as any)[key], `config.process[0].train.${key}`); setJobConfig((defaultCompileOptions as any)[key], `config.process[0].model.${key}`);
} }
} else { } else {
for (const key in defaultCompileOptions) { for (const key in defaultCompileOptions) {
setJobConfig(undefined, `config.process[0].train.${key}`); setJobConfig(undefined, `config.process[0].model.${key}`);
} }
} }
}} }}

12
ui/src/types.ts
View File

@@ -151,12 +151,6 @@ export interface TrainConfig {
differential_guidance_scale?: number; differential_guidance_scale?: number;
audio_loss_multiplier?: number; audio_loss_multiplier?: number;
max_loss?: number | null; max_loss?: number | null;
compile?: boolean;
block_compile?: boolean;
compile_mode?: 'default' | 'max-autotune' | 'fastest';
compile_fullgraph?: boolean;
compile_dynamic?: boolean;
cache_size_limit?: number;
} }
export interface QuantizeKwargsConfig { export interface QuantizeKwargsConfig {
@@ -177,6 +171,12 @@ export interface ModelConfig {
layer_offloading_transformer_percent?: number; layer_offloading_transformer_percent?: number;
layer_offloading_text_encoder_percent?: number; layer_offloading_text_encoder_percent?: number;
assistant_lora_path?: string; assistant_lora_path?: string;
compile?: boolean;
block_compile?: boolean;
compile_mode?: 'default' | 'max-autotune' | 'fastest';
compile_fullgraph?: boolean;
compile_dynamic?: boolean;
cache_size_limit?: number;
} }
export interface SampleItem { export interface SampleItem {

2
version.py
View File

@@ -1 +1 @@
VERSION = "0.10.7" VERSION = "0.10.8"