Hugging Face's logo

Spaces:
ZhouZJ36DL
/
Multi-turn_Consistent_Image_Editing_FLUX.1-dev
Running on Zero

App Files Community
ZhouZJ36DL commited on
Commit
72ddbe6
·
1 Parent(s): 7b9ea65

modified: src/flux/sampling.py

Browse files
Files changed (10) hide show
  1. src/flux/__pycache__/__init__.cpython-310.pyc +0 -0
  2. src/flux/__pycache__/_version.cpython-310.pyc +0 -0
  3. src/flux/__pycache__/math.cpython-310.pyc +0 -0
  4. src/flux/__pycache__/model.cpython-310.pyc +0 -0
  5. src/flux/__pycache__/sampling.cpython-310.pyc +0 -0
  6. src/flux/__pycache__/util.cpython-310.pyc +0 -0
  7. src/flux/modules/__pycache__/autoencoder.cpython-310.pyc +0 -0
  8. src/flux/modules/__pycache__/conditioner.cpython-310.pyc +0 -0
  9. src/flux/modules/__pycache__/layers.cpython-310.pyc +0 -0
  10. src/flux/sampling.py +15 -16
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
 
src/flux/sampling.py CHANGED
@@ -522,22 +522,21 @@ def denoise_multi_turn_consistent(
522
  info['second_order'] = False
523
  info['inject'] = inject_list[i]
524
 
525
- # Check and print devices of all input tensors
526
- tensor_info = {
527
- 'img': img,
528
- 'img_ids': img_ids,
529
- 'txt': txt,
530
- 'txt_ids': txt_ids,
531
- 'y (vec)': vec, # Renamed for clarity in print
532
- 'timesteps (t_vec)': t_vec, # Renamed for clarity in print
533
- 'guidance (guidance_vec)': guidance_vec # Renamed for clarity in print
534
- }
535
- for name, tensor in tensor_info.items():
536
- if torch.is_tensor(tensor):
537
- print(f"{name} device: {tensor.device}, shape: {tensor.shape}")
538
- else:
539
- print(f"{name} is not a tensor: {type(tensor)} (value: {tensor})")
540
-
541
 
542
 
543
  if next_step_velocity is None:
 
522
  info['second_order'] = False
523
  info['inject'] = inject_list[i]
524
 
525
+ # Check and print devices of all input tensors
526
+ tensor_info = {
527
+ 'img': img,
528
+ 'img_ids': img_ids,
529
+ 'txt': txt,
530
+ 'txt_ids': txt_ids,
531
+ 'y (vec)': vec, # Renamed for clarity in print
532
+ 'timesteps (t_vec)': t_vec, # Renamed for clarity in print
533
+ 'guidance (guidance_vec)': guidance_vec # Renamed for clarity in print
534
+ }
535
+ for name, tensor in tensor_info.items():
536
+ if torch.is_tensor(tensor):
537
+ print(f"{name} device: {tensor.device}, shape: {tensor.shape}")
538
+ else:
539
+ print(f"{name} is not a tensor: {type(tensor)} (value: {tensor})")
 
540
 
541
 
542
  if next_step_velocity is None: