Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
93bc0f4
1
Parent(s):
e615641
modified: app.py
Browse files- app.py +6 -7
- src/flux/__pycache__/__init__.cpython-310.pyc +0 -0
- src/flux/__pycache__/_version.cpython-310.pyc +0 -0
- src/flux/__pycache__/math.cpython-310.pyc +0 -0
- src/flux/__pycache__/model.cpython-310.pyc +0 -0
- src/flux/__pycache__/sampling.cpython-310.pyc +0 -0
- src/flux/__pycache__/util.cpython-310.pyc +0 -0
- src/flux/modules/__pycache__/autoencoder.cpython-310.pyc +0 -0
- src/flux/modules/__pycache__/conditioner.cpython-310.pyc +0 -0
- src/flux/modules/__pycache__/layers.cpython-310.pyc +0 -0
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
|
|
|