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:
|