Unverified Commit e6037e82 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

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

update llamatokenizer in hunyuanvideo tests
parent 196aef5a
...@@ -132,7 +132,7 @@ class HunyuanVideoPipelineFastTests(PipelineTesterMixin, PyramidAttentionBroadca ...@@ -132,7 +132,7 @@ class HunyuanVideoPipelineFastTests(PipelineTesterMixin, PyramidAttentionBroadca
torch.manual_seed(0) torch.manual_seed(0)
text_encoder = LlamaModel(llama_text_encoder_config) 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) torch.manual_seed(0)
text_encoder_2 = CLIPTextModel(clip_text_encoder_config) text_encoder_2 = CLIPTextModel(clip_text_encoder_config)
...@@ -155,10 +155,8 @@ class HunyuanVideoPipelineFastTests(PipelineTesterMixin, PyramidAttentionBroadca ...@@ -155,10 +155,8 @@ class HunyuanVideoPipelineFastTests(PipelineTesterMixin, PyramidAttentionBroadca
else: else:
generator = torch.Generator(device=device).manual_seed(seed) 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 = { inputs = {
"prompt": "", "prompt": "dance monkey",
"prompt_template": { "prompt_template": {
"template": "{}", "template": "{}",
"crop_start": 0, "crop_start": 0,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment