Yannic Kilcher commited on
Commit
a74df5c
1 Parent(s): 99b07a7
Files changed (2) hide show
  1. interface.py +2 -2
  2. interface_projector.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
 
interface_projector.py CHANGED
@@ -27,7 +27,7 @@ else:
27
 
28
  def _new_forward(self, *args, **kwargs):
29
  kwargs["force_fp32"] = True
30
- return _old_forward(self, *args, **kwargs)
31
 
32
  G.forward = types.MethodType(_new_forward, G)
33
 
@@ -35,7 +35,7 @@ else:
35
 
36
  def _new_synthesis_forward(self, *args, **kwargs):
37
  kwargs["force_fp32"] = True
38
- return _old_synthesis_forward(self, *args, **kwargs)
39
 
40
  G.synthesis.forward = types.MethodType(_new_synthesis_forward, G.synthesis)
41
 
27
 
28
  def _new_forward(self, *args, **kwargs):
29
  kwargs["force_fp32"] = True
30
+ return _old_forward(*args, **kwargs)
31
 
32
  G.forward = types.MethodType(_new_forward, G)
33
 
35
 
36
  def _new_synthesis_forward(self, *args, **kwargs):
37
  kwargs["force_fp32"] = True
38
+ return _old_synthesis_forward(*args, **kwargs)
39
 
40
  G.synthesis.forward = types.MethodType(_new_synthesis_forward, G.synthesis)
41