John6666 commited on
Commit
76d770d
1 Parent(s): 31ff6a8

Upload dc.py

Browse files
Files changed (1) hide show
  1. dc.py +2 -2
dc.py CHANGED
@@ -140,11 +140,11 @@ flux_pipe = FluxPipeline.from_pretrained(
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:
 
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: