Boese0601 commited on
Commit
453bd9c
·
verified ·
1 Parent(s): c03b667

Update src/flux/xflux_pipeline.py

Browse files
Files changed (1) hide show
  1. src/flux/xflux_pipeline.py +1 -1
src/flux/xflux_pipeline.py CHANGED
@@ -388,7 +388,7 @@ class XFluxPipeline:
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
- super().__init__()
392
  self.device = device
393
  self.controlnet_loaded = controlnet_loaded
394
  self.ip_loaded = ip_loaded
 
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
+ super().__init__(model_type="flux-dev", device=device, offload=False)
392
  self.device = device
393
  self.controlnet_loaded = controlnet_loaded
394
  self.ip_loaded = ip_loaded