Spaces:
Running
on
Zero
Running
on
Zero
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
|
|
|