multimodalart HF Staff commited on
Commit
d91fd4c
·
verified ·
1 Parent(s): 20db3d7

Update R3GAN/FusedOperators.py

Browse files
Files changed (1) hide show
  1. R3GAN/FusedOperators.py +1 -1
R3GAN/FusedOperators.py CHANGED
@@ -30,4 +30,4 @@ class BiasedActivationCUDA(nn.Module):
30
  def forward(self, x):
31
  return bias_act.bias_act(x, self.Bias.to(x.dtype), act=BiasedActivationCUDA.Function, gain=1)
32
 
33
- BiasedActivation = BiasedActivationCUDA
 
30
  def forward(self, x):
31
  return bias_act.bias_act(x, self.Bias.to(x.dtype), act=BiasedActivationCUDA.Function, gain=1)
32
 
33
+ BiasedActivation = BiasedActivationReference