use dream8

pulid/pipeline.py

@@ -37,7 +37,7 @@ class PuLIDPipeline:
         self.device = 'cuda'
         
         # SG161222/RealVisXL_V5.0_Lightning, lykon/dreamshaper-xl-lightning
-        self.pipe = AutoPipelineForText2Image.from_pretrained('SG161222/RealVisXL_V5.0_Lightning', torch_dtype=torch.float16, variant="fp16")
+        self.pipe = AutoPipelineForText2Image.from_pretrained('lykon/dreamshaper-xl-lightning', torch_dtype=torch.float16, variant="fp16")
         self.hack_unet_attn_layers(self.pipe.unet)
         self.pipe = self.pipe.to(self.device)