github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[NFC] polish applications/Chat/coati/models/utils.py codestyle (#4277) 

* [NFC] polish colossalai/context/random/__init__.py code style

* [NFC] polish applications/Chat/coati/models/utils.py code style
pull/4338/head
yuxuan-lou 2023-07-19 22:18:30 +08:00 committed by binmakeswell
parent 9e512938f6
commit 0991405361
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
applications/Chat/coati/models/utils.py
View File

@ -46,10 +46,7 @@ def log_probs_from_logits(logits: torch.Tensor, labels: torch.Tensor) -> torch.T
return log_probs_labels.squeeze(-1) return log_probs_labels.squeeze(-1)
def calc_action_log_probs(output: torch.Tensor, def calc_action_log_probs(output: torch.Tensor, sequences: torch.LongTensor, num_actions: int) -> torch.Tensor:
sequences: torch.LongTensor,
num_actions: int
) -> torch.Tensor:
"""Calculate action log probs. """Calculate action log probs.
Args: Args: