ZhouZJ36DL commited on
Commit
dcb641e
·
1 Parent(s): ae529cc

modified: src/flux/math.py

Browse files
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__/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/math.py CHANGED
@@ -111,7 +111,7 @@ def scaled_dot_product_attention(query, key, value, txt_shape, img_shape, cur_st
111
  attn_weight = torch.softmax(attn_weight, dim=-1)
112
  attn_weight = torch.dropout(attn_weight, dropout_p, train=True)
113
 
114
- '''
115
  print(f"[DEBUG_Sample] query:\n{query}")
116
  print(f"[DEBUG_Sample] key :\n{key}")
117
  print(f"[DEBUG_Sample] value :\n{value}")
 
111
  attn_weight = torch.softmax(attn_weight, dim=-1)
112
  attn_weight = torch.dropout(attn_weight, dropout_p, train=True)
113
 
114
+ '''
115
  print(f"[DEBUG_Sample] query:\n{query}")
116
  print(f"[DEBUG_Sample] key :\n{key}")
117
  print(f"[DEBUG_Sample] value :\n{value}")
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