Update

TheSDEs · Dec 3, 2024 · a624c34 · a624c34
1 parent 168e908
commit a624c34
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/hyvideo/utils/preprocess_text_encoder_tokenizer_utils.py b/hyvideo/utils/preprocess_text_encoder_tokenizer_utils.py
@@ -3,13 +3,12 @@
 from transformers import (
     AutoProcessor,
     LlavaForConditionalGeneration,
-    LlavaProcessor
 )
 
 
 def preprocess_text_encoder_tokenizer(args):
 
-    processor: LlavaProcessor = AutoProcessor.from_pretrained(args.input_dir)
+    processor = AutoProcessor.from_pretrained(args.input_dir)
     model = LlavaForConditionalGeneration.from_pretrained(
         args.input_dir,
         torch_dtype=torch.float16,