Det the dataset dropout for yue2
This commit is contained in:
1 parent
7bc91e91a5
commit
900c0e8a67
2 files changed
+10
No files matched your search
@@ -1420,6 +1420,14 @@ export default function SimpleJob({
|
||||
// automaticallt add the controls for a new dataset
|
||||
const controls = modelArch?.controls ?? [];
|
||||
newDataset.controls = controls;
|
||||
// arch dataset defaults (datasets[x].*) apply to added datasets too, not just at arch switch
|
||||
for (const key in modelArch?.defaults ?? {}) {
|
||||
const marker = 'datasets[x].';
|
||||
const idx = key.indexOf(marker);
|
||||
if (idx !== -1) {
|
||||
(newDataset as any)[key.slice(idx + marker.length)] = modelArch!.defaults![key][0];
|
||||
}
|
||||
}
|
||||
setJobConfig([...jobConfig.config.process[0].datasets, newDataset], 'config.process[0].datasets');
|
||||
}}
|
||||
className="w-full px-4 py-2 bg-gray-700 hover:bg-gray-600 rounded-lg transition-colors"
|
||||
|
||||
@@ -1392,6 +1392,8 @@ export const modelArchs: ModelArch[] = [
|
||||
'config.process[0].model.qtype': ['qfloat8', 'qfloat8'],
|
||||
'config.process[0].sample': [defaultYue2SampleConfig, defaultSampleConfig],
|
||||
'config.process[0].datasets[x].cache_latents_to_disk': [true, true],
|
||||
// blank captions break lyric following; the AR must always see the prefix
|
||||
'config.process[0].datasets[x].caption_dropout_rate': [0, 0.05],
|
||||
},
|
||||
// native YuE2 prompt: style text, a [Lyrics] line, the lyrics
|
||||
hasMultiLinePrompts: true,
|
||||
|
||||
Reference in new issue
Block a user