Update utils.py
Browse files
utils.py
CHANGED
|
@@ -181,7 +181,6 @@ def save_attention_maps(attn_maps, tokenizer, prompts, base_dir='attn_maps', unc
|
|
| 181 |
os.mkdir(batch_dir)
|
| 182 |
|
| 183 |
startofword = True
|
| 184 |
-
print('tokens',tokens)
|
| 185 |
for i, (token, a) in enumerate(zip(tokens, attn_map[:len(tokens)])):
|
| 186 |
if '</w>' in token:
|
| 187 |
token = token.replace('</w>', '')
|
|
|
|
| 181 |
os.mkdir(batch_dir)
|
| 182 |
|
| 183 |
startofword = True
|
|
|
|
| 184 |
for i, (token, a) in enumerate(zip(tokens, attn_map[:len(tokens)])):
|
| 185 |
if '</w>' in token:
|
| 186 |
token = token.replace('</w>', '')
|