Update README.md

README.md

@@ -21,30 +21,28 @@ To use `AudSemThinker-QA-GRPO` for audio question answering, you can load it usi
 
 ```python
 import soundfile as sf
-from transformers import Qwen2_5OmniForConditionalGeneration, Qwen2_5OmniProcessor
+from transformers import Qwen2_5OmniThinkerForConditionalGeneration, Qwen2_5OmniProcessor
 from qwen_omni_utils import process_mm_info
 import torchaudio
 
 # default: Load the model on the available device(s)
-model = Qwen2_5OmniForConditionalGeneration.from_pretrained(
-    "gijs/audsemthinker-qa-grpo", 
+model = Qwen2_5OmniThinkerForConditionalGeneration.from_pretrained(
+    "gijs/audsemthinker", 
     torch_dtype="auto", 
     device_map="auto",
-    trust_remote_code=True,
-    low_cpu_mem_usage=True
+    trust_remote_code=True
 )
 
 # We recommend enabling flash_attention_2 for better acceleration and memory saving.
-# model = Qwen2_5OmniForConditionalGeneration.from_pretrained(
-#     "gijs/audsemthinker-qa-grpo",
+# model = Qwen2_5OmniThinkerForConditionalGeneration.from_pretrained(
+#     "gijs/audsemthinker",
 #     torch_dtype="auto",
 #     device_map="auto",
 #     attn_implementation="flash_attention_2",
-#     trust_remote_code=True,
-#     low_cpu_mem_usage=True
+#     trust_remote_code=True
 # )
 
-processor = Qwen2_5OmniProcessor.from_pretrained("gijs/audsemthinker-qa-grpo", trust_remote_code=True)
+processor = Qwen2_5OmniProcessor.from_pretrained("Qwen/Qwen2.5-Omni-7B", trust_remote_code=True)
 
 # Load and preprocess audio
 audio_file = "path/to/your/audio.wav"
@@ -79,7 +77,7 @@ conversation = [
 
 # Preparation for inference
 text = processor.apply_chat_template(conversation, add_generation_prompt=True, tokenize=False)
-audios, images, videos = process_mm_info(conversation)
+audios, images, videos = process_mm_info(conversation, use_audio_in_video=False)
 inputs = processor(
     text=text, 
     audio=audios,