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