ZhouZJ36DL commited on
Commit
93bc0f4
·
1 Parent(s): e615641

modified: app.py

Browse files
app.py CHANGED
@@ -72,13 +72,6 @@ class FluxEditor:
72
  self.ae.eval()
73
  self.model.eval()
74
 
75
- print(f"Inital_t5_device: {self.t5.hf_module.device}")
76
- print(f"Inital_clip_device: {self.clip.hf_module.device}")
77
- print(f"Inital_flow_model: {self.model.img_in.weight.device}")
78
- print(f"Inital_flow_model self.model.img_in: {self.model.img_in}")
79
- print(f"Inital_flow_model self.model.time_in.out_layer.weight: {self.model.time_in.out_layer.weight}")
80
-
81
-
82
  # clear history
83
  if os.path.exists("history_gradio/history.safetensors"):
84
  os.remove("history_gradio/history.safetensors")
@@ -260,6 +253,12 @@ class FluxEditor:
260
 
261
  torch.cuda.empty_cache()
262
  seed = None
 
 
 
 
 
 
263
 
264
  if self.offload:
265
  self.model.cpu()
 
72
  self.ae.eval()
73
  self.model.eval()
74
 
 
 
 
 
 
 
 
75
  # clear history
76
  if os.path.exists("history_gradio/history.safetensors"):
77
  os.remove("history_gradio/history.safetensors")
 
253
 
254
  torch.cuda.empty_cache()
255
  seed = None
256
+
257
+ print(f"Inital_t5_device: {self.t5.hf_module.device}")
258
+ print(f"Inital_clip_device: {self.clip.hf_module.device}")
259
+ print(f"Inital_flow_model: {self.model.img_in.weight.device}")
260
+ print(f"Inital_flow_model self.model.img_in: {self.model.img_in}")
261
+ print(f"Inital_flow_model self.model.time_in.out_layer.weight: {self.model.time_in.out_layer.weight}")
262
 
263
  if self.offload:
264
  self.model.cpu()
src/flux/__pycache__/__init__.cpython-310.pyc CHANGED
Binary files a/src/flux/__pycache__/__init__.cpython-310.pyc and b/src/flux/__pycache__/__init__.cpython-310.pyc differ
 
src/flux/__pycache__/_version.cpython-310.pyc CHANGED
Binary files a/src/flux/__pycache__/_version.cpython-310.pyc and b/src/flux/__pycache__/_version.cpython-310.pyc differ
 
src/flux/__pycache__/math.cpython-310.pyc CHANGED
Binary files a/src/flux/__pycache__/math.cpython-310.pyc and b/src/flux/__pycache__/math.cpython-310.pyc differ
 
src/flux/__pycache__/model.cpython-310.pyc CHANGED
Binary files a/src/flux/__pycache__/model.cpython-310.pyc and b/src/flux/__pycache__/model.cpython-310.pyc differ
 
src/flux/__pycache__/sampling.cpython-310.pyc CHANGED
Binary files a/src/flux/__pycache__/sampling.cpython-310.pyc and b/src/flux/__pycache__/sampling.cpython-310.pyc differ
 
src/flux/__pycache__/util.cpython-310.pyc CHANGED
Binary files a/src/flux/__pycache__/util.cpython-310.pyc and b/src/flux/__pycache__/util.cpython-310.pyc differ
 
src/flux/modules/__pycache__/autoencoder.cpython-310.pyc CHANGED
Binary files a/src/flux/modules/__pycache__/autoencoder.cpython-310.pyc and b/src/flux/modules/__pycache__/autoencoder.cpython-310.pyc differ
 
src/flux/modules/__pycache__/conditioner.cpython-310.pyc CHANGED
Binary files a/src/flux/modules/__pycache__/conditioner.cpython-310.pyc and b/src/flux/modules/__pycache__/conditioner.cpython-310.pyc differ
 
src/flux/modules/__pycache__/layers.cpython-310.pyc CHANGED
Binary files a/src/flux/modules/__pycache__/layers.cpython-310.pyc and b/src/flux/modules/__pycache__/layers.cpython-310.pyc differ