Skip to content

Commit e6037e8

Browse files
authored
[tests] update llamatokenizer in hunyuanvideo tests (huggingface#10681)
update llamatokenizer in hunyuanvideo tests
1 parent 196aef5 commit e6037e8

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/pipelines/hunyuan_video/test_hunyuan_video.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ def get_dummy_components(self, num_layers: int = 1, num_single_layers: int = 1):
132132

133133
torch.manual_seed(0)
134134
text_encoder = LlamaModel(llama_text_encoder_config)
135-
tokenizer = LlamaTokenizer.from_pretrained("hf-internal-testing/tiny-random-LlamaForCausalLM")
135+
tokenizer = LlamaTokenizer.from_pretrained("finetrainers/dummy-hunyaunvideo", subfolder="tokenizer")
136136

137137
torch.manual_seed(0)
138138
text_encoder_2 = CLIPTextModel(clip_text_encoder_config)
@@ -155,10 +155,8 @@ def get_dummy_inputs(self, device, seed=0):
155155
else:
156156
generator = torch.Generator(device=device).manual_seed(seed)
157157

158-
# Cannot test with dummy prompt because tokenizers are not configured correctly.
159-
# TODO(aryan): create dummy tokenizers and using from hub
160158
inputs = {
161-
"prompt": "",
159+
"prompt": "dance monkey",
162160
"prompt_template": {
163161
"template": "{}",
164162
"crop_start": 0,

0 commit comments

Comments
 (0)