ToTheBeginning / PuLID

Official code for PuLID: Pure and Lightning ID Customization via Contrastive Alignment
Apache License 2.0
946 stars 59 forks source link

Please change base model to latest fine tuned sdxl model like juggernaut x or any other model #12

Open putuoka opened 2 months ago

putuoka commented 2 months ago

pipeline.py

class PuLIDPipeline:
    def __init__(self, *args, **kwargs):
        super().__init__()
        self.device = 'cuda'
        sdxl_base_repo = 'stabilityai/stable-diffusion-xl-base-1.0'
        sdxl_lightning_repo = 'ByteDance/SDXL-Lightning'
        self.sdxl_base_repo = sdxl_base_repo
brentjohnston commented 2 months ago

Totally agree, i was able to change it in my last comment here but it was a huge pain and the IDencoder weights had mismatches, had to use diffusers version of model I trained. Used claude to work around the weights issue.

Juggnernaut x would have been me much better, lora support in gradio demo, 8 step lightning or no lightning option, and comfyui node.