Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -195,9 +195,9 @@ else:
|
|
| 195 |
mem.append(
|
| 196 |
int(
|
| 197 |
torch.cuda.get_device_properties(i).total_memory
|
| 198 |
-
/
|
| 199 |
-
/
|
| 200 |
-
/
|
| 201 |
+ 0.4
|
| 202 |
)
|
| 203 |
)
|
|
|
|
| 195 |
mem.append(
|
| 196 |
int(
|
| 197 |
torch.cuda.get_device_properties(i).total_memory
|
| 198 |
+
/ 16384
|
| 199 |
+
/ 16384
|
| 200 |
+
/ 16384
|
| 201 |
+ 0.4
|
| 202 |
)
|
| 203 |
)
|