Skip to content

Commit f1b7a5a

Browse files
committed
tweaked yue to improve autofit
1 parent 7863721 commit f1b7a5a

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

‎koboldcpp.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1955,11 +1955,11 @@ def autoset_gpu_layers(ctxsize, sdquanted, bbs, musiclowvram): #shitty algo to d
19551955
if modelfile_extracted_meta[8] > 1024*1024*10: #embeddings model tax
19561956
calulated_gpu_overhead += max(350*1024*1024, modelfile_extracted_meta[8] * 1.5)
19571957
if modelfile_extracted_meta[9] > 1024*1024*10: #music llm tax
1958-
musicoh1 = modelfile_extracted_meta[9] * 1.05 + (300*1024*1024)
1958+
musicoh1 = modelfile_extracted_meta[9] * 1.1 + (400*1024*1024)
19591959
if modelfile_extracted_meta[10] > 1024*1024*10: #music dit tax
1960-
musicoh2 = modelfile_extracted_meta[10] * 1.05 + (300*1024*1024)
1960+
musicoh2 = modelfile_extracted_meta[10] * 1.1 + (400*1024*1024)
19611961
if modelfile_extracted_meta[11] > 1024*1024*10: #music vae tax
1962-
musicoh3 = modelfile_extracted_meta[11] * 1.05
1962+
musicoh3 = modelfile_extracted_meta[11] * 1.1
19631963
if musiclowvram:
19641964
calulated_gpu_overhead += max(musicoh1,musicoh2,musicoh3)
19651965
else:

‎otherarch/audiocpp/upstream/include/engine/models/yue2/types.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ struct Yue2VAEConfig {
3838
int64_t latent_dim = 64;
3939
int64_t encoder_latent_dim = 128;
4040
int64_t downsampling_ratio = 1920;
41-
int64_t decode_core_frames = 1024;
41+
int64_t decode_core_frames = 512;
4242
int64_t decode_halo_frames = 16;
4343
};
4444

0 commit comments

Comments
 (0)