fix

Author: yiyixuxu

src/diffusers/pipelines/hunyuan_image/pipeline_hunyuanimage.py

@@ -20,7 +20,7 @@
 import torch
 from transformers import ByT5Tokenizer, Qwen2_5_VLForConditionalGeneration, Qwen2Tokenizer, T5EncoderModel
 
-from ...guiders import AdaptiveProjectedMixGuidance, ClassifierFreeGuidance
+from ...guiders import AdaptiveProjectedMixGuidance
 from ...image_processor import VaeImageProcessor
 from ...models import AutoencoderKLHunyuanImage, HunyuanImageTransformer2DModel
 from ...schedulers import FlowMatchEulerDiscreteScheduler
@@ -199,7 +199,7 @@ def __init__(
         text_encoder_2: T5EncoderModel,
         tokenizer_2: ByT5Tokenizer,
         transformer: HunyuanImageTransformer2DModel,
-        guider: Union[AdaptiveProjectedMixGuidance, ClassifierFreeGuidance],
+        guider: AdaptiveProjectedMixGuidance,
         ocr_guider: Optional[AdaptiveProjectedMixGuidance] = None,
     ):
         super().__init__()

src/diffusers/pipelines/hunyuan_image/pipeline_hunyuanimage_refiner.py

@@ -19,7 +19,7 @@
 import torch
 from transformers import Qwen2_5_VLForConditionalGeneration, Qwen2Tokenizer
 
-from ...guiders import ClassifierFreeGuidance
+from ...guiders import AdaptiveProjectedMixGuidance
 from ...image_processor import PipelineImageInput, VaeImageProcessor
 from ...models import AutoencoderKLHunyuanImageRefiner, HunyuanImageTransformer2DModel
 from ...schedulers import FlowMatchEulerDiscreteScheduler
@@ -163,7 +163,7 @@ def __init__(
         text_encoder: Qwen2_5_VLForConditionalGeneration,
         tokenizer: Qwen2Tokenizer,
         transformer: HunyuanImageTransformer2DModel,
-        guider: ClassifierFreeGuidance,
+        guider: AdaptiveProjectedMixGuidance,
     ):
         super().__init__()