Spaces:
Running
on
Zero
Running
on
Zero
Update src/flux/xflux_pipeline.py
Browse files
src/flux/xflux_pipeline.py
CHANGED
@@ -387,12 +387,7 @@ class XFluxPipeline:
|
|
387 |
|
388 |
|
389 |
class XFluxSampler(XFluxPipeline):
|
390 |
-
def __init__(self,
|
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
|