Boese0601 commited on
Commit
80c1007
·
verified ·
1 Parent(s): 8373499

Update src/flux/xflux_pipeline.py

Browse files
Files changed (1) hide show
  1. src/flux/xflux_pipeline.py +1 -6
src/flux/xflux_pipeline.py CHANGED
@@ -387,12 +387,7 @@ class XFluxPipeline:
387
 
388
 
389
  class XFluxSampler(XFluxPipeline):
390
- def __init__(self, clip, t5, ae, model, device,controlnet_loaded=False,ip_loaded=False, spatial_condition=False, offload=False, clip_image_processor=None, image_encoder=None, improj=None, share_position_embedding=False, use_share_weight_referencenet=False, single_block_refnet=False, double_block_refnet=False):
391
- self.clip = clip
392
- self.t5 = t5
393
- self.ae = ae
394
- self.model = model
395
- self.model.eval()
396
  self.device = device
397
  self.controlnet_loaded = controlnet_loaded
398
  self.ip_loaded = ip_loaded
 
387
 
388
 
389
  class XFluxSampler(XFluxPipeline):
390
+ def __init__(self, device, controlnet_loaded=False,ip_loaded=False, spatial_condition=False, offload=False, clip_image_processor=None, image_encoder=None, improj=None, share_position_embedding=False, use_share_weight_referencenet=False, single_block_refnet=False, double_block_refnet=False):
 
 
 
 
 
391
  self.device = device
392
  self.controlnet_loaded = controlnet_loaded
393
  self.ip_loaded = ip_loaded