Update pulid/pipeline.py

pulid/pipeline.py

@@ -45,6 +45,7 @@ class PuLIDPipeline:
                 hf_hub_download(sdxl_lightning_repo, 'sdxl_lightning_4step_unet.safetensors'), device=self.device
             )
         )
+        unet.half()
         self.hack_unet_attn_layers(unet)
         self.pipe = StableDiffusionXLPipeline.from_pretrained(
             sdxl_base_repo, unet=unet, torch_dtype=torch.float16, variant="fp16"
@@ -217,7 +218,7 @@ class PuLIDPipeline:
         return torch.cat((uncond_id_embedding, id_embedding), dim=0)
 
     def inference(self, prompt, size, prompt_n='', image_embedding=None, id_scale=1.0, guidance_scale=1.2, steps=4):
-        print(self.pipe.unet.time_embedding.linear_1.weight.data.dtype)
+        print(self.pipe.unet.time_embedding.linear_1.weight)
         images = self.pipe(
             prompt=prompt,
             negative_prompt=prompt_n,