Tobias Bergmann commited on
Commit
922a633
1 Parent(s): 7614629

remove dead comments

Browse files
Files changed (1) hide show
  1. pulid/pipeline.py +0 -16
pulid/pipeline.py CHANGED
@@ -35,22 +35,6 @@ class PuLIDPipeline:
35
  def __init__(self, *args, **kwargs):
36
  super().__init__()
37
  self.device = 'cuda'
38
- #sdxl_base_repo = 'stabilityai/stable-diffusion-xl-base-1.0'
39
- #sdxl_lightning_repo = 'ByteDance/SDXL-Lightning'
40
- #self.sdxl_base_repo = sdxl_base_repo
41
-
42
- # load base model
43
- #unet = UNet2DConditionModel.from_config(sdxl_base_repo, subfolder='unet').to(self.device, torch.float16)
44
- #unet.load_state_dict(
45
- # load_file(
46
- # hf_hub_download(sdxl_lightning_repo, 'sdxl_lightning_4step_unet.safetensors'), device=self.device
47
- # )
48
- #)
49
- #unet.half()
50
- #self.hack_unet_attn_layers(unet)
51
- #self.pipe = StableDiffusionXLPipeline.from_pretrained(
52
- # sdxl_base_repo, unet=unet, torch_dtype=torch.float16, variant="fp16"
53
- #).to(self.device)
54
 
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")
 
35
  def __init__(self, *args, **kwargs):
36
  super().__init__()
37
  self.device = 'cuda'
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
38
 
39
  # SG161222/RealVisXL_V5.0_Lightning, lykon/dreamshaper-xl-lightning
40
  self.pipe = AutoPipelineForText2Image.from_pretrained('SG161222/RealVisXL_V5.0_Lightning', torch_dtype=torch.float16, variant="fp16")