demo 1
Browse files
app.py
CHANGED
@@ -15,6 +15,7 @@ class SimpleModel(nn.Module):
|
|
15 |
def forward(self, x):
|
16 |
pass
|
17 |
|
|
|
18 |
def device():
|
19 |
return next(model.parameters()).device
|
20 |
|
@@ -22,8 +23,6 @@ model = SimpleModel().to(device);
|
|
22 |
|
23 |
@torch.inference_mode()
|
24 |
def sysinfo():
|
25 |
-
RandomTensor = torch.randn(1, 2) # Example audio tensor
|
26 |
-
tensorExample = RandomTensor.to(device)
|
27 |
|
28 |
return f"""
|
29 |
hostname: {platform.node()} {socket.gethostname()}
|
|
|
15 |
def forward(self, x):
|
16 |
pass
|
17 |
|
18 |
+
@property
|
19 |
def device():
|
20 |
return next(model.parameters()).device
|
21 |
|
|
|
23 |
|
24 |
@torch.inference_mode()
|
25 |
def sysinfo():
|
|
|
|
|
26 |
|
27 |
return f"""
|
28 |
hostname: {platform.node()} {socket.gethostname()}
|