[tests] update llamatokenizer in hunyuanvideo tests (#10681)

update llamatokenizer in hunyuanvideo tests
huggingface · Jan 29, 2025 · e6037e8 · e6037e8
1 parent 196aef5
commit e6037e8
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/pipelines/hunyuan_video/test_hunyuan_video.py b/tests/pipelines/hunyuan_video/test_hunyuan_video.py
@@ -132,7 +132,7 @@ def get_dummy_components(self, num_layers: int = 1, num_single_layers: int = 1):
 
         torch.manual_seed(0)
         text_encoder = LlamaModel(llama_text_encoder_config)
-        tokenizer = LlamaTokenizer.from_pretrained("hf-internal-testing/tiny-random-LlamaForCausalLM")
+        tokenizer = LlamaTokenizer.from_pretrained("finetrainers/dummy-hunyaunvideo", subfolder="tokenizer")
 
         torch.manual_seed(0)
         text_encoder_2 = CLIPTextModel(clip_text_encoder_config)
@@ -155,10 +155,8 @@ def get_dummy_inputs(self, device, seed=0):
         else:
             generator = torch.Generator(device=device).manual_seed(seed)
 
-        # Cannot test with dummy prompt because tokenizers are not configured correctly.
-        # TODO(aryan): create dummy tokenizers and using from hub
         inputs = {
-            "prompt": "",
+            "prompt": "dance monkey",
             "prompt_template": {
                 "template": "{}",
                 "crop_start": 0,