fix

asomoza · asomoza · commit 087a6f00aafa · 2025-03-20T11:54:21.000-03:00
diff --git a/tests/pipelines/cogview4/test_cogview4.py b/tests/pipelines/cogview4/test_cogview4.py
@@ -89,8 +89,7 @@ def get_dummy_components(self):
             hidden_size=32, intermediate_size=8, num_hidden_layers=2, num_attention_heads=4, head_dim=8
         )
         text_encoder = GlmForCausalLM(text_encoder_config)
-        # TODO(aryan): change this to THUDM/CogView4 once released
-        tokenizer = AutoTokenizer.from_pretrained("THUDM/glm-4-9b-chat", trust_remote_code=True)
+        tokenizer = AutoTokenizer.from_pretrained("THUDM/CogView4-6B", subfolder="tokenizer")
 
         components = {
             "transformer": transformer,

Original file line number	Diff line number	Diff line change
`@@ -89,8 +89,7 @@ def get_dummy_components(self):`
`89`	`89`	`hidden_size=32, intermediate_size=8, num_hidden_layers=2, num_attention_heads=4, head_dim=8`
`90`	`90`	`)`
`91`	`91`	`text_encoder = GlmForCausalLM(text_encoder_config)`
`92`		`- # TODO(aryan): change this to THUDM/CogView4 once released`
`93`		`- tokenizer = AutoTokenizer.from_pretrained("THUDM/glm-4-9b-chat", trust_remote_code=True)`
	`92`	`+ tokenizer = AutoTokenizer.from_pretrained("THUDM/CogView4-6B", subfolder="tokenizer")`
`94`	`93`
`95`	`94`	`components = {`
`96`	`95`	`"transformer": transformer,`