Yannic Kilcher commited on
Commit
e42078a
1 Parent(s): f29d636
Files changed (1) hide show
  1. interface.py +2 -2
interface.py CHANGED
@@ -22,7 +22,7 @@ else:
22
 
23
  def _new_forward(self, *args, **kwargs):
24
  kwargs["force_fp32"] = True
25
- return _old_forward(self, *args, **kwargs)
26
 
27
  G.forward = types.MethodType(_new_forward, G)
28
 
@@ -30,7 +30,7 @@ else:
30
 
31
  def _new_synthesis_forward(self, *args, **kwargs):
32
  kwargs["force_fp32"] = True
33
- return _old_synthesis_forward(self, *args, **kwargs)
34
 
35
  G.synthesis.forward = types.MethodType(_new_synthesis_forward, G.synthesis)
36
 
 
22
 
23
  def _new_forward(self, *args, **kwargs):
24
  kwargs["force_fp32"] = True
25
+ return _old_forward(*args, **kwargs)
26
 
27
  G.forward = types.MethodType(_new_forward, G)
28
 
 
30
 
31
  def _new_synthesis_forward(self, *args, **kwargs):
32
  kwargs["force_fp32"] = True
33
+ return _old_synthesis_forward(*args, **kwargs)
34
 
35
  G.synthesis.forward = types.MethodType(_new_synthesis_forward, G.synthesis)
36