Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
898d15e
1
Parent(s):
bad7f63
modified: app.py
Browse files- app.py +1 -1
- 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
|
@@ -204,9 +204,9 @@ class FluxEditor:
|
|
| 204 |
torch.cuda.empty_cache()
|
| 205 |
self.ae.decoder.to(x.device)
|
| 206 |
|
| 207 |
-
self.ae = self.ae.to(x.device)
|
| 208 |
# decode latents to pixel space
|
| 209 |
x = unpack(x[0].float(), height, width)
|
|
|
|
| 210 |
with torch.autocast(device_type=self.device.type, dtype=torch.bfloat16):
|
| 211 |
x = self.ae.decode(x)
|
| 212 |
|
|
|
|
| 204 |
torch.cuda.empty_cache()
|
| 205 |
self.ae.decoder.to(x.device)
|
| 206 |
|
|
|
|
| 207 |
# decode latents to pixel space
|
| 208 |
x = unpack(x[0].float(), height, width)
|
| 209 |
+
self.ae = self.ae.to(x.device)
|
| 210 |
with torch.autocast(device_type=self.device.type, dtype=torch.bfloat16):
|
| 211 |
x = self.ae.decode(x)
|
| 212 |
|
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
|
|
|