Spaces:
Running
on
Zero
Running
on
Zero
Upload dc.py
Browse files
dc.py
CHANGED
|
@@ -142,8 +142,6 @@ flux_pipe = FluxPipeline.from_pretrained(
|
|
| 142 |
torch_dtype=torch.bfloat16,
|
| 143 |
)#.to("cuda")
|
| 144 |
components = flux_pipe.components
|
| 145 |
-
components.pop("transformer", None)
|
| 146 |
-
components.pop("scheduler", None)
|
| 147 |
delete_model(flux_repo)
|
| 148 |
#components = None
|
| 149 |
|
|
|
|
| 142 |
torch_dtype=torch.bfloat16,
|
| 143 |
)#.to("cuda")
|
| 144 |
components = flux_pipe.components
|
|
|
|
|
|
|
| 145 |
delete_model(flux_repo)
|
| 146 |
#components = None
|
| 147 |
|