Spaces:

ZhouZJ36DL
/

Multi-turn_Consistent_Image_Editing_FLUX.1-dev

Running on Zero

ZhouZJ36DL commited on Aug 19

Commit

ae529cc

1 Parent(s): 5960a43

modified: src/flux/math.py

Files changed (6) hide show

src/flux/__pycache__/__init__.cpython-310.pyc CHANGED Viewed

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 Viewed

Binary files a/src/flux/__pycache__/_version.cpython-310.pyc and b/src/flux/__pycache__/_version.cpython-310.pyc differ

src/flux/__pycache__/model.cpython-310.pyc CHANGED Viewed

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 Viewed

Binary files a/src/flux/__pycache__/sampling.cpython-310.pyc and b/src/flux/__pycache__/sampling.cpython-310.pyc differ

src/flux/math.py CHANGED Viewed

@@ -111,17 +111,12 @@ def scaled_dot_product_attention(query, key, value, txt_shape, img_shape, cur_st
     attn_weight = torch.softmax(attn_weight, dim=-1)
     attn_weight = torch.dropout(attn_weight, dropout_p, train=True)
-   ''' if info['inverse']:
-        print(f"[DEBUG_Inverse] query:\n{query}")
-        print(f"[DEBUG_Inverse] key :\n{key}")
-        print(f"[DEBUG_Inverse] value :\n{value}")
-        print(f"[DEBUG_Inverse] attn_weight:\n{attn_weight}")
-    else:
-        print(f"[DEBUG_Sample] query:\n{query}")
-        print(f"[DEBUG_Sample] key :\n{key}")
-        print(f"[DEBUG_Sample] value :\n{value}")
-        print(f"[DEBUG_Sample] attn_weight values:\n{attn_weight}")
-'''
     if not info['inverse']:
         # GENERATE MASK
         txt_img_cross = attn_weight[:, :, -img_shape:, :txt_shape]  # lower left part

     attn_weight = torch.softmax(attn_weight, dim=-1)
     attn_weight = torch.dropout(attn_weight, dropout_p, train=True)
+   '''
+    print(f"[DEBUG_Sample] query:\n{query}")
+    print(f"[DEBUG_Sample] key :\n{key}")
+    print(f"[DEBUG_Sample] value :\n{value}")
+    print(f"[DEBUG_Sample] attn_weight values:\n{attn_weight}")
+    '''
     if not info['inverse']:
         # GENERATE MASK
         txt_img_cross = attn_weight[:, :, -img_shape:, :txt_shape]  # lower left part

src/flux/modules/__pycache__/layers.cpython-310.pyc CHANGED Viewed

Binary files a/src/flux/modules/__pycache__/layers.cpython-310.pyc and b/src/flux/modules/__pycache__/layers.cpython-310.pyc differ