Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -121,8 +121,6 @@ flux_pipe = FluxPipeline.from_pretrained(
|
|
| 121 |
torch_dtype=torch.bfloat16,
|
| 122 |
).to("cuda")
|
| 123 |
components = flux_pipe.components
|
| 124 |
-
components.pop("transformer", None)
|
| 125 |
-
components.pop("scheduler", None)
|
| 126 |
delete_model(flux_repo)
|
| 127 |
# components = None
|
| 128 |
|
|
|
|
| 121 |
torch_dtype=torch.bfloat16,
|
| 122 |
).to("cuda")
|
| 123 |
components = flux_pipe.components
|
|
|
|
|
|
|
| 124 |
delete_model(flux_repo)
|
| 125 |
# components = None
|
| 126 |
|