mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-10 04:18:05 +00:00
fix Tensor is not defined (#4129)
This commit is contained in:
parent
7e46bc87b6
commit
8abc87798f
@ -1,4 +1,5 @@
|
||||
import torch
|
||||
from torch import Tensor
|
||||
|
||||
|
||||
def bias_dropout_add(x, bias, residual, prob, training):
|
||||
|
Loading…
Reference in New Issue
Block a user