mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-11 13:59:08 +00:00
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
This commit is contained in:
@@ -11,7 +11,6 @@ def exists(val):
|
||||
|
||||
|
||||
def eval_decorator(fn):
|
||||
|
||||
def inner(model, *args, **kwargs):
|
||||
was_training = model.training
|
||||
model.eval()
|
||||
@@ -34,7 +33,6 @@ def top_k(logits, thres=0.9):
|
||||
|
||||
|
||||
class AutoregressiveWrapper(nn.Module):
|
||||
|
||||
def __init__(self, net, max_seq_len=2048, pad_value=0):
|
||||
super().__init__()
|
||||
self.max_seq_len = max_seq_len
|
||||
|
Reference in New Issue
Block a user