Update pulid/pipeline.py
Browse files- pulid/pipeline.py +1 -1
pulid/pipeline.py
CHANGED
@@ -55,7 +55,7 @@ class PuLIDPipeline:
|
|
55 |
# SG161222/RealVisXL_V5.0_Lightning, lykon/dreamshaper-xl-lightning
|
56 |
self.pipe = AutoPipelineForText2Image.from_pretrained('SG161222/RealVisXL_V5.0_Lightning', torch_dtype=torch.float16, variant="fp16")
|
57 |
self.hack_unet_attn_layers(self.pipe.unet)
|
58 |
-
pipe = pipe.to(self.device)
|
59 |
|
60 |
self.pipe.watermark = None
|
61 |
|
|
|
55 |
# SG161222/RealVisXL_V5.0_Lightning, lykon/dreamshaper-xl-lightning
|
56 |
self.pipe = AutoPipelineForText2Image.from_pretrained('SG161222/RealVisXL_V5.0_Lightning', torch_dtype=torch.float16, variant="fp16")
|
57 |
self.hack_unet_attn_layers(self.pipe.unet)
|
58 |
+
self.pipe = self.pipe.to(self.device)
|
59 |
|
60 |
self.pipe.watermark = None
|
61 |
|