from hyimage.diffusion.pipelines.hunyuanimage_pipeline import HunyuanImagePipeline pipe = HunyuanImagePipeline.from_pretrained(model_name=self.model_id, torch_dtype='bf16', device=device) pipe.to(device) 这样好像不行
from hyimage.diffusion.pipelines.hunyuanimage_pipeline import HunyuanImagePipeline
pipe = HunyuanImagePipeline.from_pretrained(model_name=self.model_id, torch_dtype='bf16', device=device)
pipe.to(device)
这样好像不行