update

Author: lvyuanjun.lyj

tests/models/qwen2_5_omni/test_modeling_qwen2_5_omni.py

@@ -728,7 +728,7 @@ def test_resize_embeddings_untied_with_deepspeed(self):
 @require_torch
 class Qwen2_5OmniModelIntegrationTest(unittest.TestCase):
     def setUp(self):
-        self.processor = AutoProcessor.from_pretrained("/home/xj_data/lvyuanjun.lyj/res/navit_thinker_talker/test")
+        self.processor = AutoProcessor.from_pretrained("Qwen/Qwen2.5-Omni-7B")
         self.audio_url = "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen2-Audio/audio/glass-breaking-151256.mp3"
         self.audio_url_additional = (
             "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen2-Audio/audio/f2641_0_throatclearing.wav"
@@ -758,9 +758,7 @@ def tearDown(self):
 
     @slow
     def test_small_model_integration_test(self):
-        model = Qwen2_5OmniModel.from_pretrained(
-            "/home/xj_data/lvyuanjun.lyj/res/navit_thinker_talker/test", torch_dtype=torch.float32, device_map="auto"
-        )
+        model = Qwen2_5OmniModel.from_pretrained("Qwen/Qwen2.5-Omni-7B", torch_dtype=torch.float32, device_map="auto")
 
         text = self.processor.apply_chat_template(self.messages, tokenize=False, add_generation_prompt=True)
         inputs = self.processor(
@@ -818,9 +816,7 @@ def test_small_model_integration_test(self):
 
     @slow
     def test_small_model_integration_test_batch(self):
-        model = Qwen2_5OmniModel.from_pretrained(
-            "/home/xj_data/lvyuanjun.lyj/res/navit_thinker_talker/test", torch_dtype=torch.float32, device_map="auto"
-        )
+        model = Qwen2_5OmniModel.from_pretrained("Qwen/Qwen2.5-Omni-7B", torch_dtype=torch.float32, device_map="auto")
         text = self.processor.apply_chat_template(self.messages, tokenize=False, add_generation_prompt=True)
         inputs = self.processor(
             text=[text, text],
@@ -844,9 +840,7 @@ def test_small_model_integration_test_batch(self):
 
     @slow
     def test_small_model_integration_test_multiturn(self):
-        model = Qwen2_5OmniModel.from_pretrained(
-            "/home/xj_data/lvyuanjun.lyj/res/navit_thinker_talker/test", torch_dtype=torch.float32, device_map="auto"
-        )
+        model = Qwen2_5OmniModel.from_pretrained("Qwen/Qwen2.5-Omni-7B", torch_dtype=torch.float32, device_map="auto")
 
         messages = [
             self.messages[0],
@@ -883,9 +877,7 @@ def test_small_model_integration_test_multiturn(self):
 
     @slow
     def test_small_model_integration_test_w_audio(self):
-        model = Qwen2_5OmniModel.from_pretrained(
-            "/home/xj_data/lvyuanjun.lyj/res/navit_thinker_talker/test", torch_dtype=torch.float32, device_map="auto"
-        )
+        model = Qwen2_5OmniModel.from_pretrained("Qwen/Qwen2.5-Omni-7B", torch_dtype=torch.float32, device_map="auto")
         audio_url = "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen2-Audio/audio/guess_age_gender.wav"
 
         messages = [
@@ -918,7 +910,7 @@ def test_small_model_integration_test_w_audio(self):
     @require_torch_gpu
     def test_small_model_integration_test_batch_flashatt2(self):
         model = Qwen2_5OmniModel.from_pretrained(
-            "/home/xj_data/lvyuanjun.lyj/res/navit_thinker_talker/test",
+            "Qwen/Qwen2.5-Omni-7B",
             torch_dtype=torch.bfloat16,
             attn_implementation="flash_attention_2",
             device_map="auto",