Change hidream to be proper x0 prediction and use a custom weighing scheme for x0 models.

This commit is contained in:
Jaret Burkett committed 2026-09-10 11:49:54 -06:00
1 parent b1bf3e4381
commit f56b5a1d40
5 files changed
+1035 -20

No files matched your search

+1 -1
View File
@@ -1419,7 +1419,7 @@ export const modelArchs: ModelArch[] = [
'config.process[0].model.name_or_path': ['HiDream-ai/HiDream-O1-Image', defaultNameOrPath],
'config.process[0].model.quantize': [true, false],
'config.process[0].model.quantize_te': [false, false],
'config.process[0].train.timestep_type': ['linear', 'sigmoid'],
'config.process[0].train.timestep_type': ['weighted', 'sigmoid'],
'config.process[0].network.conv': [undefined, 16],
'config.process[0].network.conv_alpha': [undefined, 16],
'config.process[0].train.max_loss': [1.0, undefined],