lglg666
/

SongGeneration-large

Model card Files Files and versions

lglg666 commited on Oct 15

Commit

75e2043

·

verified ·

1 Parent(s): 386e6f9

Update config.yaml

Files changed (1) hide show

config.yaml +1 -1

config.yaml CHANGED Viewed

@@ -121,7 +121,7 @@ offload:
     clean_cache_wrapper:
       module: self
       method_name: _sample_next_token
-      diff_mem_gb_thre: 2
     debug: false
   wav_tokenizer_diffusion:

     clean_cache_wrapper:
       module: self
       method_name: _sample_next_token
+      diff_mem_gb_thre: 5
     debug: false
   wav_tokenizer_diffusion: