Skip to content

Commit 87915f9

Browse files
committed
chore
Signed-off-by: guochenxu <[email protected]>
1 parent 3f8eebc commit 87915f9

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

tests/models/minicpm_o_2_6/test_modeling_minicpm_o_2_6.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,6 @@ def setUp(self):
7373
f.write(chunk)
7474

7575
self.model_path = "openbmb/MiniCPM-o-2_6"
76-
# self.model_path = "/cache/guochenxu/models/minicpmo2_6_renamed_qkv"
7776
self.model = AutoModel.from_pretrained(
7877
self.model_path,
7978
attn_implementation='sdpa',

tests/models/minicpm_o_2_6/test_processing_minicpm_o_2_6.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def setUpClass(cls):
6060
cls.tmpdirname = tempfile.mkdtemp()
6161
processor_kwargs = cls.prepare_processor_dict()
6262
processor = AutoProcessor.from_pretrained(
63-
"/cache/guochenxu/models/minicpmo2_6_renamed_qkv", **processor_kwargs)
63+
"openbmb/MiniCPM-o-2_6", **processor_kwargs)
6464
processor.save_pretrained(cls.tmpdirname)
6565

6666
@staticmethod

0 commit comments

Comments
 (0)