Spaces:
Running on Zero
Running on Zero
Update R3GAN/FusedOperators.py
Browse files- 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 =
|
|
|
|
| 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
|