Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "1f73609adb358aa1a004f6e5e4c0928a13eb93be"
Commit 09914053 authored by yuxuan-lou's avatar yuxuan-lou Committed by binmakeswell
Browse files

[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
parent 9e512938
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 4 deletions
applications/Chat/coati/models/utils.py
View file @ 09914053
...@@ -46,10 +46,7 @@ def log_probs_from_logits(logits: torch.Tensor, labels: torch.Tensor) -> torch.T ...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment