Tobias Bergmann
commited on
Commit
·
922a633
1
Parent(s):
7614629
remove dead comments
Browse files- 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")
|