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
|