Spaces:
Runtime error
Runtime error
Балаганский Никита Николаевич
commited on
Commit
•
99fe246
1
Parent(s):
8f9df38
fix
Browse files- sampling.py +1 -0
sampling.py
CHANGED
|
@@ -53,6 +53,7 @@ class CAIFSampler:
|
|
| 53 |
target_cls_id: int = 0,
|
| 54 |
**kwargs
|
| 55 |
):
|
|
|
|
| 56 |
next_token_logits = output_logis[:, -1]
|
| 57 |
next_token_log_probs = F.log_softmax(
|
| 58 |
next_token_logits, dim=-1
|
|
|
|
| 53 |
target_cls_id: int = 0,
|
| 54 |
**kwargs
|
| 55 |
):
|
| 56 |
+
print(act_type)
|
| 57 |
next_token_logits = output_logis[:, -1]
|
| 58 |
next_token_log_probs = F.log_softmax(
|
| 59 |
next_token_logits, dim=-1
|