Update README.md

README.md

@@ -25,11 +25,12 @@ margin. Our work reveals that LLMs can be an excellent compressor for music, but
 
 <!-- <audio controls src="https://cdn-uploads.huggingface.co/production/uploads/5fd6f670053c8345eddc1b68/8NSONUjIF7KGUCfwzPCd9.mpga"></audio> -->
 
-<iframe width="787" height="528" src="https://www.youtube.com/embed/zt3l49K55Io" title="ChatMusician: Fostering Intrinsic Musical Abilities Into LLM" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe>
+<video src="https://www.youtube.com/embed/zt3l49K55Io" width="320" height="240" controls></video>
+<!-- <iframe width="787" height="528" src="https://www.youtube.com/embed/zt3l49K55Io" title="ChatMusician: Fostering Intrinsic Musical Abilities Into LLM" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe>
 
 ## Usage
 
-You can use the models through Huggingface's Transformers library. Check our Github repo for more advanced use: [https://github.com/hf-lin/ChatMusician](https://github.com/hf-lin/ChatMusician)
+You can use the models through Huggingface's Transformers library. Check our Github repo for more advanced use: [https://github.com/hf-lin/ChatMusician](https://github.com/hf-lin/ChatMusician) -->
 
 ## CLI demo
 ```python
@@ -37,8 +38,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig
 from string import Template
 prompt_template = Template("Human: ${inst} </s> Assistant: ")
 
-tokenizer = AutoTokenizer.from_pretrained("m-a-p/ChatMusician-v1-sft-78k", trust_remote_code=True)
-model = AutoModelForCausalLM.from_pretrained("m-a-p/ChatMusician-v1-sft-78k", trust_remote_code=True).eval()
+tokenizer = AutoTokenizer.from_pretrained("m-a-p/ChatMusician", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("m-a-p/ChatMusician", trust_remote_code=True).eval()
 model.cuda()
 generation_config = GenerationConfig(
     temperature=0.2,