Spaces:
Running
Running
Yannic Kilcher
commited on
Commit
•
e42078a
1
Parent(s):
f29d636
bugfix
Browse files- 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(
|
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(
|
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 |
|