TobDeBer commited on
Commit
b16d8a7
·
verified ·
1 Parent(s): d8db4fe

Update pulid/pipeline.py

Browse files
Files changed (1) hide show
  1. 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