Spaces:
Running
on
T4
Running
on
T4
try to figure out how ZeroGPU works
Browse files
app.py
CHANGED
@@ -35,6 +35,7 @@ class ControllableInterface(torch.nn.Module):
|
|
35 |
self.model.device = "cpu"
|
36 |
self.wgan.to("cpu")
|
37 |
self.wgan.device = "cpu"
|
|
|
38 |
|
39 |
def read(self,
|
40 |
prompt,
|
|
|
35 |
self.model.device = "cpu"
|
36 |
self.wgan.to("cpu")
|
37 |
self.wgan.device = "cpu"
|
38 |
+
self._modules = []
|
39 |
|
40 |
def read(self,
|
41 |
prompt,
|