File tree Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Expand file tree Collapse file tree 2 files changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -72,12 +72,12 @@ def test_llama3(self):
7272 self .check_block_names (block_names , ["model.layers" ], [32 ])
7373 assert is_pure_text_model (model )
7474
75- def test_mixtral (self ):
76- model_name = "/models/Mixtral-8x7B-Instruct-v0.1"
77- model = AutoModelForCausalLM .from_pretrained (model_name , torch_dtype = "auto" , trust_remote_code = True )
78- block_names = get_block_names (model )
79- self .check_block_names (block_names , ["model.layers" ], [32 ])
80- assert is_pure_text_model (model )
75+ # def test_mixtral(self):
76+ # model_name = "/models/Mixtral-8x7B-Instruct-v0.1"
77+ # model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", trust_remote_code=True)
78+ # block_names = get_block_names(model)
79+ # self.check_block_names(block_names, ["model.layers"], [32])
80+ # assert is_pure_text_model(model)
8181
8282 def test_falcon (self ):
8383 model_name = "/models/Falcon3-7B-Instruct"
Original file line number Diff line number Diff line change @@ -199,7 +199,7 @@ def test_vlm_gguf(self):
199199 autoround .quantize_and_save (output_dir = quantized_model_path , format = "gguf:q4_0" )
200200 self .assertTrue ("mmproj-model.gguf" in os .listdir ("./saved" ))
201201 file_size = os .path .getsize ("./saved/Qwen2.5-VL-7B-Instruct-Q4_0.gguf" ) / 1024 ** 2
202- self .assertAlmostEqual (file_size , 4226 , delta = 5.0 )
202+ self .assertAlmostEqual (file_size , 4242 , delta = 5.0 )
203203 file_size = os .path .getsize ("./saved/mmproj-model.gguf" ) / 1024 ** 2
204204 self .assertAlmostEqual (file_size , 2580 , delta = 5.0 )
205205 shutil .rmtree ("./saved" , ignore_errors = True )
You can’t perform that action at this time.
0 commit comments