From b0b74ea02b892c356cdd5336457ef83dd466bacb Mon Sep 17 00:00:00 2001 From: smthemex <138738845+smthemex@users.noreply.github.com> Date: Sun, 22 Sep 2024 16:55:02 +0800 Subject: [PATCH] Update pipeline_flux.py --- PuLID/pulid/pipeline_flux.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/PuLID/pulid/pipeline_flux.py b/PuLID/pulid/pipeline_flux.py index 0296f56..88490a1 100644 --- a/PuLID/pulid/pipeline_flux.py +++ b/PuLID/pulid/pipeline_flux.py @@ -109,10 +109,11 @@ def img2tensor(imgs, bgr2rgb=True, float32=True): class PuLIDPipeline(nn.Module): - def __init__(self, dit, device, weight_dtype=torch.bfloat16, *args, **kwargs): + def __init__(self, dit, device,clip_vision_path, weight_dtype=torch.bfloat16, *args, **kwargs): super().__init__() self.device = device self.weight_dtype = weight_dtype + self.clip_vision_path=clip_vision_path double_interval = 2 single_interval = 4 @@ -145,7 +146,7 @@ class PuLIDPipeline(nn.Module): self.face_helper.face_parse = None self.face_helper.face_parse = init_parsing_model(model_name='bisenet', device=self.device) # clip-vit backbone - model, _, _ = create_model_and_transforms('EVA02-CLIP-L-14-336', 'eva_clip', force_custom_clip=True) + model, _, _ = create_model_and_transforms('EVA02-CLIP-L-14-336', clip_vision_path, force_custom_clip=True) model = model.visual self.clip_vision_model = model.to(self.device, dtype=self.weight_dtype) eva_transform_mean = getattr(self.clip_vision_model, 'image_mean', OPENAI_DATASET_MEAN)