Update README.md

README.md

@@ -57,7 +57,7 @@ elastic_mode = "S"
 
 prompt = "A groovy funk bassline with a tight drum beat"
 output_wav_path = "generated_audio_elastic_S.wav"
-hf_token = "YOUR_HF_TOKEN" #
+hf_token = "YOUR_HF_TOKEN"
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 processor = AutoProcessor.from_pretrained(model_name_hf, token=hf_token)
@@ -65,7 +65,7 @@ processor = AutoProcessor.from_pretrained(model_name_hf, token=hf_token)
 model = MusicgenForConditionalGeneration.from_pretrained(
     model_name_hf,
     token=hf_token,
-    torch_dtype=torch.float16, # Or float32, matching compilation
+    torch_dtype=torch.float16,
     mode=elastic_mode,
     device=device,
     __full_patch=True,