Spaces:
Sleeping
Sleeping
Hugo Flores Garcia
commited on
Commit
•
e3c7f46
1
Parent(s):
84d4ed6
c2f can be None
Browse files- vampnet/interface.py +2 -0
vampnet/interface.py
CHANGED
|
@@ -48,6 +48,8 @@ class Interface:
|
|
| 48 |
self.c2f.to(device)
|
| 49 |
self.c2f.eval()
|
| 50 |
self.c2f.chunk_size_s = self.s2t2s(coarse2fine_chunk_size_s)
|
|
|
|
|
|
|
| 51 |
|
| 52 |
self.device = device
|
| 53 |
|
|
|
|
| 48 |
self.c2f.to(device)
|
| 49 |
self.c2f.eval()
|
| 50 |
self.c2f.chunk_size_s = self.s2t2s(coarse2fine_chunk_size_s)
|
| 51 |
+
else:
|
| 52 |
+
self.c2f = None
|
| 53 |
|
| 54 |
self.device = device
|
| 55 |
|