Update metaVoice.py

metaVoice.py

@@ -137,7 +137,7 @@ class Model:
         tokenizer_cls: Type[TrainedBPETokeniser],
         decoder_cls: Type[Decoder],
         data_adapter_fn,
-        use_kv_cache: None,
+        use_kv_cache= None,
     ):
         # TODO: disentangle the encodec stuff and numbers etc with rest of this code (esp at encoder-only / second stage model inference)
         # TODO: remove magic number
@@ -724,7 +724,7 @@ class SamplingControllerConfig:
     init_from: str = "resume"
     """Either 'resume' (from an out_dir) or a gpt2 variant (e.g. 'gpt2-xl')."""
 
-    use_kv_cache: None
+    use_kv_cache: Optional[None] = None
     # """Type of kv caching to use for inference: 1) [none] no kv caching, 2) [flash_decoding] use the 
     # flash decoding kernel, 3) [vanilla] use torch attention with hand implemented kv-cache."""