ZhouZJ36DL commited on
Commit
898d15e
·
1 Parent(s): bad7f63

modified: app.py

Browse files
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