deepsafe commited on
Commit
acbba5e
·
verified ·
1 Parent(s): 3680c22

sync from GitHub (3150c97)

Browse files
audio/aasist3/api.py CHANGED
@@ -84,7 +84,7 @@ if DEVICE.type == "cuda":
84
  logger.info(
85
  "Device: cuda (%s, %.1f GB VRAM)",
86
  torch.cuda.get_device_name(0),
87
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
88
  )
89
  else:
90
  logger.warning(
@@ -175,7 +175,7 @@ def _gpu_health_info() -> dict:
175
  "gpu_name": torch.cuda.get_device_name(0),
176
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
177
  "vram_total_mb": round(
178
- torch.cuda.get_device_properties(0).total_mem / 1024**2
179
  ),
180
  }
181
  return {}
 
84
  logger.info(
85
  "Device: cuda (%s, %.1f GB VRAM)",
86
  torch.cuda.get_device_name(0),
87
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
88
  )
89
  else:
90
  logger.warning(
 
175
  "gpu_name": torch.cuda.get_device_name(0),
176
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
177
  "vram_total_mb": round(
178
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
179
  ),
180
  }
181
  return {}
audio/nes2net/api.py CHANGED
@@ -97,7 +97,7 @@ if DEVICE.type == "cuda":
97
  logger.info(
98
  "Device: cuda (%s, %.1f GB VRAM)",
99
  torch.cuda.get_device_name(0),
100
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
101
  )
102
  else:
103
  logger.warning(
@@ -198,7 +198,7 @@ def _gpu_health_info() -> dict:
198
  "gpu_name": torch.cuda.get_device_name(0),
199
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
200
  "vram_total_mb": round(
201
- torch.cuda.get_device_properties(0).total_mem / 1024**2
202
  ),
203
  }
204
  return {}
 
97
  logger.info(
98
  "Device: cuda (%s, %.1f GB VRAM)",
99
  torch.cuda.get_device_name(0),
100
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
101
  )
102
  else:
103
  logger.warning(
 
198
  "gpu_name": torch.cuda.get_device_name(0),
199
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
200
  "vram_total_mb": round(
201
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
202
  ),
203
  }
204
  return {}
audio/safeear/api.py CHANGED
@@ -81,7 +81,7 @@ if DEVICE.type == "cuda":
81
  logger.info(
82
  "Device: cuda (%s, %.1f GB VRAM)",
83
  torch.cuda.get_device_name(0),
84
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
85
  )
86
  else:
87
  logger.warning(
@@ -234,7 +234,7 @@ def _gpu_health_info() -> dict:
234
  "gpu_name": torch.cuda.get_device_name(0),
235
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
236
  "vram_total_mb": round(
237
- torch.cuda.get_device_properties(0).total_mem / 1024**2
238
  ),
239
  }
240
  return {}
 
81
  logger.info(
82
  "Device: cuda (%s, %.1f GB VRAM)",
83
  torch.cuda.get_device_name(0),
84
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
85
  )
86
  else:
87
  logger.warning(
 
234
  "gpu_name": torch.cuda.get_device_name(0),
235
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
236
  "vram_total_mb": round(
237
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
238
  ),
239
  }
240
  return {}
audio/shiftyspeech/api.py CHANGED
@@ -97,7 +97,7 @@ if DEVICE.type == "cuda":
97
  logger.info(
98
  "Device: cuda (%s, %.1f GB VRAM)",
99
  torch.cuda.get_device_name(0),
100
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
101
  )
102
  else:
103
  logger.warning(
@@ -197,7 +197,7 @@ def _gpu_health_info() -> dict:
197
  "gpu_name": torch.cuda.get_device_name(0),
198
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
199
  "vram_total_mb": round(
200
- torch.cuda.get_device_properties(0).total_mem / 1024**2
201
  ),
202
  }
203
  return {}
 
97
  logger.info(
98
  "Device: cuda (%s, %.1f GB VRAM)",
99
  torch.cuda.get_device_name(0),
100
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
101
  )
102
  else:
103
  logger.warning(
 
197
  "gpu_name": torch.cuda.get_device_name(0),
198
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
199
  "vram_total_mb": round(
200
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
201
  ),
202
  }
203
  return {}
audio/sonics/app.py CHANGED
@@ -78,7 +78,7 @@ if DEVICE.type == "cuda":
78
  logger.info(
79
  "Device: cuda (%s, %.1f GB VRAM)",
80
  torch.cuda.get_device_name(0),
81
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
82
  )
83
  else:
84
  logger.warning(
@@ -165,7 +165,7 @@ def _gpu_health_info() -> dict:
165
  "gpu_name": torch.cuda.get_device_name(0),
166
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
167
  "vram_total_mb": round(
168
- torch.cuda.get_device_properties(0).total_mem / 1024**2
169
  ),
170
  }
171
  return {}
 
78
  logger.info(
79
  "Device: cuda (%s, %.1f GB VRAM)",
80
  torch.cuda.get_device_name(0),
81
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
82
  )
83
  else:
84
  logger.warning(
 
165
  "gpu_name": torch.cuda.get_device_name(0),
166
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
167
  "vram_total_mb": round(
168
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
169
  ),
170
  }
171
  return {}
image/aide/app.py CHANGED
@@ -91,7 +91,7 @@ if DEVICE.type == "cuda":
91
  logger.info(
92
  "Device: cuda (%s, %.1f GB VRAM)",
93
  torch.cuda.get_device_name(0),
94
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
95
  )
96
  else:
97
  logger.warning(
@@ -321,7 +321,7 @@ def _gpu_health_info() -> dict:
321
  "gpu_name": torch.cuda.get_device_name(0),
322
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
323
  "vram_total_mb": round(
324
- torch.cuda.get_device_properties(0).total_mem / 1024**2
325
  ),
326
  }
327
  return {}
 
91
  logger.info(
92
  "Device: cuda (%s, %.1f GB VRAM)",
93
  torch.cuda.get_device_name(0),
94
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
95
  )
96
  else:
97
  logger.warning(
 
321
  "gpu_name": torch.cuda.get_device_name(0),
322
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
323
  "vram_total_mb": round(
324
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
325
  ),
326
  }
327
  return {}
image/cospy/app.py CHANGED
@@ -114,7 +114,7 @@ if DEVICE.type == "cuda":
114
  logger.info(
115
  "Device: cuda (%s, %.1f GB VRAM)",
116
  torch.cuda.get_device_name(0),
117
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
118
  )
119
  else:
120
  logger.warning(
@@ -332,7 +332,7 @@ def _gpu_health_info() -> dict:
332
  "gpu_name": torch.cuda.get_device_name(0),
333
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
334
  "vram_total_mb": round(
335
- torch.cuda.get_device_properties(0).total_mem / 1024**2
336
  ),
337
  }
338
  return {}
 
114
  logger.info(
115
  "Device: cuda (%s, %.1f GB VRAM)",
116
  torch.cuda.get_device_name(0),
117
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
118
  )
119
  else:
120
  logger.warning(
 
332
  "gpu_name": torch.cuda.get_device_name(0),
333
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
334
  "vram_total_mb": round(
335
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
336
  ),
337
  }
338
  return {}
image/effort/app.py CHANGED
@@ -115,7 +115,7 @@ if DEVICE.type == "cuda":
115
  logger.info(
116
  "Device: cuda (%s, %.1f GB VRAM)",
117
  torch.cuda.get_device_name(0),
118
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
119
  )
120
  else:
121
  logger.warning(
@@ -382,7 +382,7 @@ def _gpu_health_info() -> dict:
382
  "gpu_name": torch.cuda.get_device_name(0),
383
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
384
  "vram_total_mb": round(
385
- torch.cuda.get_device_properties(0).total_mem / 1024**2
386
  ),
387
  }
388
  return {}
 
115
  logger.info(
116
  "Device: cuda (%s, %.1f GB VRAM)",
117
  torch.cuda.get_device_name(0),
118
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
119
  )
120
  else:
121
  logger.warning(
 
382
  "gpu_name": torch.cuda.get_device_name(0),
383
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
384
  "vram_total_mb": round(
385
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
386
  ),
387
  }
388
  return {}
image/fsd/app.py CHANGED
@@ -84,7 +84,7 @@ if DEVICE == "cuda":
84
  logger.info(
85
  "Device: cuda (%s, %.1f GB VRAM)",
86
  torch.cuda.get_device_name(0),
87
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
88
  )
89
  else:
90
  logger.warning(
@@ -255,7 +255,7 @@ def _gpu_health_info() -> dict:
255
  "gpu_name": torch.cuda.get_device_name(0),
256
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
257
  "vram_total_mb": round(
258
- torch.cuda.get_device_properties(0).total_mem / 1024**2
259
  ),
260
  }
261
  return {}
 
84
  logger.info(
85
  "Device: cuda (%s, %.1f GB VRAM)",
86
  torch.cuda.get_device_name(0),
87
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
88
  )
89
  else:
90
  logger.warning(
 
255
  "gpu_name": torch.cuda.get_device_name(0),
256
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
257
  "vram_total_mb": round(
258
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
259
  ),
260
  }
261
  return {}
image/npr/app.py CHANGED
@@ -113,7 +113,7 @@ if DEVICE.type == "cuda":
113
  logger.info(
114
  "Device: cuda (%s, %.1f GB VRAM)",
115
  torch.cuda.get_device_name(0),
116
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
117
  )
118
  else:
119
  logger.warning(
@@ -311,7 +311,7 @@ def _gpu_health_info() -> dict:
311
  "gpu_name": torch.cuda.get_device_name(0),
312
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
313
  "vram_total_mb": round(
314
- torch.cuda.get_device_properties(0).total_mem / 1024**2
315
  ),
316
  }
317
  return {}
 
113
  logger.info(
114
  "Device: cuda (%s, %.1f GB VRAM)",
115
  torch.cuda.get_device_name(0),
116
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
117
  )
118
  else:
119
  logger.warning(
 
311
  "gpu_name": torch.cuda.get_device_name(0),
312
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
313
  "vram_total_mb": round(
314
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
315
  ),
316
  }
317
  return {}
image/universal/app.py CHANGED
@@ -75,7 +75,7 @@ if DEVICE.type == "cuda":
75
  logger.info(
76
  "Device: cuda (%s, %.1f GB VRAM)",
77
  torch.cuda.get_device_name(0),
78
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
79
  )
80
  else:
81
  logger.warning(
@@ -352,7 +352,7 @@ def _gpu_health_info() -> dict:
352
  "gpu_name": torch.cuda.get_device_name(0),
353
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
354
  "vram_total_mb": round(
355
- torch.cuda.get_device_properties(0).total_mem / 1024**2
356
  ),
357
  }
358
  return {}
 
75
  logger.info(
76
  "Device: cuda (%s, %.1f GB VRAM)",
77
  torch.cuda.get_device_name(0),
78
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
79
  )
80
  else:
81
  logger.warning(
 
352
  "gpu_name": torch.cuda.get_device_name(0),
353
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
354
  "vram_total_mb": round(
355
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
356
  ),
357
  }
358
  return {}
image/yermandy/app.py CHANGED
@@ -154,7 +154,7 @@ if DEVICE.type == "cuda":
154
  logger.info(
155
  "Device: cuda (%s, %.1f GB VRAM)",
156
  torch.cuda.get_device_name(0),
157
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
158
  )
159
  else:
160
  logger.warning(
@@ -326,7 +326,7 @@ def _gpu_health_info() -> dict:
326
  "gpu_name": torch.cuda.get_device_name(0),
327
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
328
  "vram_total_mb": round(
329
- torch.cuda.get_device_properties(0).total_mem / 1024**2
330
  ),
331
  }
332
  return {}
 
154
  logger.info(
155
  "Device: cuda (%s, %.1f GB VRAM)",
156
  torch.cuda.get_device_name(0),
157
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
158
  )
159
  else:
160
  logger.warning(
 
326
  "gpu_name": torch.cuda.get_device_name(0),
327
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
328
  "vram_total_mb": round(
329
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
330
  ),
331
  }
332
  return {}
provenance/audioseal/app.py CHANGED
@@ -81,7 +81,7 @@ if DEVICE.type == "cuda":
81
  logger.info(
82
  "Device: cuda (%s, %.1f GB VRAM)",
83
  torch.cuda.get_device_name(0),
84
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
85
  )
86
  else:
87
  logger.warning(
@@ -227,7 +227,7 @@ def _gpu_health_info() -> dict:
227
  "gpu_name": torch.cuda.get_device_name(0),
228
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
229
  "vram_total_mb": round(
230
- torch.cuda.get_device_properties(0).total_mem / 1024**2
231
  ),
232
  }
233
  return {}
 
81
  logger.info(
82
  "Device: cuda (%s, %.1f GB VRAM)",
83
  torch.cuda.get_device_name(0),
84
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
85
  )
86
  else:
87
  logger.warning(
 
227
  "gpu_name": torch.cuda.get_device_name(0),
228
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
229
  "vram_total_mb": round(
230
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
231
  ),
232
  }
233
  return {}
provenance/videoseal/app.py CHANGED
@@ -86,7 +86,7 @@ if DEVICE.type == "cuda":
86
  logger.info(
87
  "Device: cuda (%s, %.1f GB VRAM)",
88
  torch.cuda.get_device_name(0),
89
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
90
  )
91
  else:
92
  logger.warning(
@@ -363,7 +363,7 @@ def _gpu_health_info() -> dict:
363
  "gpu_name": torch.cuda.get_device_name(0),
364
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
365
  "vram_total_mb": round(
366
- torch.cuda.get_device_properties(0).total_mem / 1024**2
367
  ),
368
  }
369
  return {}
 
86
  logger.info(
87
  "Device: cuda (%s, %.1f GB VRAM)",
88
  torch.cuda.get_device_name(0),
89
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
90
  )
91
  else:
92
  logger.warning(
 
363
  "gpu_name": torch.cuda.get_device_name(0),
364
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
365
  "vram_total_mb": round(
366
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
367
  ),
368
  }
369
  return {}
video/fake-stormer/app.py CHANGED
@@ -93,7 +93,7 @@ def _load_models():
93
  logger.info(
94
  "Device: cuda (%s, %.1f GB VRAM)",
95
  torch.cuda.get_device_name(0),
96
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
97
  )
98
  else:
99
  logger.warning(
@@ -181,7 +181,7 @@ def _gpu_health_info() -> dict:
181
  "gpu_name": torch.cuda.get_device_name(0),
182
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
183
  "vram_total_mb": round(
184
- torch.cuda.get_device_properties(0).total_mem / 1024**2
185
  ),
186
  }
187
  return {}
 
93
  logger.info(
94
  "Device: cuda (%s, %.1f GB VRAM)",
95
  torch.cuda.get_device_name(0),
96
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
97
  )
98
  else:
99
  logger.warning(
 
181
  "gpu_name": torch.cuda.get_device_name(0),
182
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
183
  "vram_total_mb": round(
184
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
185
  ),
186
  }
187
  return {}
video/lipfd/app.py CHANGED
@@ -116,7 +116,7 @@ def _load_models() -> None:
116
  logger.info(
117
  "Device: cuda (%s, %.1f GB VRAM)",
118
  torch.cuda.get_device_name(0),
119
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
120
  )
121
  else:
122
  logger.warning(
@@ -221,7 +221,7 @@ def _gpu_health_info() -> dict:
221
  torch.cuda.memory_allocated(0) / 1024**2
222
  ),
223
  "vram_total_mb": round(
224
- torch.cuda.get_device_properties(0).total_mem / 1024**2
225
  ),
226
  }
227
  return {}
 
116
  logger.info(
117
  "Device: cuda (%s, %.1f GB VRAM)",
118
  torch.cuda.get_device_name(0),
119
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
120
  )
121
  else:
122
  logger.warning(
 
221
  torch.cuda.memory_allocated(0) / 1024**2
222
  ),
223
  "vram_total_mb": round(
224
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
225
  ),
226
  }
227
  return {}
video/mintime/app.py CHANGED
@@ -205,7 +205,7 @@ def _load_models() -> None:
205
  logger.info(
206
  "Device: cuda (%s, %.1f GB VRAM)",
207
  torch.cuda.get_device_name(0),
208
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
209
  )
210
  else:
211
  logger.warning(
@@ -319,7 +319,7 @@ def _gpu_health_info() -> dict:
319
  "gpu_name": torch.cuda.get_device_name(0),
320
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
321
  "vram_total_mb": round(
322
- torch.cuda.get_device_properties(0).total_mem / 1024**2
323
  ),
324
  }
325
  return {}
 
205
  logger.info(
206
  "Device: cuda (%s, %.1f GB VRAM)",
207
  torch.cuda.get_device_name(0),
208
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
209
  )
210
  else:
211
  logger.warning(
 
319
  "gpu_name": torch.cuda.get_device_name(0),
320
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
321
  "vram_total_mb": round(
322
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
323
  ),
324
  }
325
  return {}
video/pwtf-dvd/app.py CHANGED
@@ -128,7 +128,7 @@ def _load_models() -> None:
128
  logger.info(
129
  "Device: cuda (%s, %.1f GB VRAM)",
130
  torch.cuda.get_device_name(0),
131
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
132
  )
133
  else:
134
  logger.warning(
@@ -243,7 +243,7 @@ def _gpu_health_info() -> dict:
243
  torch.cuda.memory_allocated(0) / 1024**2
244
  ),
245
  "vram_total_mb": round(
246
- torch.cuda.get_device_properties(0).total_mem / 1024**2
247
  ),
248
  }
249
  return {}
 
128
  logger.info(
129
  "Device: cuda (%s, %.1f GB VRAM)",
130
  torch.cuda.get_device_name(0),
131
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
132
  )
133
  else:
134
  logger.warning(
 
243
  torch.cuda.memory_allocated(0) / 1024**2
244
  ),
245
  "vram_total_mb": round(
246
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
247
  ),
248
  }
249
  return {}
video/recce/app.py CHANGED
@@ -130,7 +130,7 @@ def _load_models() -> None:
130
  logger.info(
131
  "Device: cuda (%s, %.1f GB VRAM)",
132
  torch.cuda.get_device_name(0),
133
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
134
  )
135
  else:
136
  logger.warning(
@@ -247,7 +247,7 @@ def _gpu_health_info() -> dict:
247
  torch.cuda.memory_allocated(0) / 1024**2
248
  ),
249
  "vram_total_mb": round(
250
- torch.cuda.get_device_properties(0).total_mem / 1024**2
251
  ),
252
  }
253
  return {}
 
130
  logger.info(
131
  "Device: cuda (%s, %.1f GB VRAM)",
132
  torch.cuda.get_device_name(0),
133
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
134
  )
135
  else:
136
  logger.warning(
 
247
  torch.cuda.memory_allocated(0) / 1024**2
248
  ),
249
  "vram_total_mb": round(
250
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
251
  ),
252
  }
253
  return {}
video/sbi/app.py CHANGED
@@ -118,7 +118,7 @@ def _load_models() -> None:
118
  logger.info(
119
  "Device: cuda (%s, %.1f GB VRAM)",
120
  torch.cuda.get_device_name(0),
121
- torch.cuda.get_device_properties(0).total_mem / 1024**3,
122
  )
123
  else:
124
  logger.warning(
@@ -211,7 +211,7 @@ def _gpu_health_info() -> dict:
211
  "gpu_name": torch.cuda.get_device_name(0),
212
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
213
  "vram_total_mb": round(
214
- torch.cuda.get_device_properties(0).total_mem / 1024**2
215
  ),
216
  }
217
  return {}
 
118
  logger.info(
119
  "Device: cuda (%s, %.1f GB VRAM)",
120
  torch.cuda.get_device_name(0),
121
+ torch.cuda.get_device_properties(0).total_memory / 1024**3,
122
  )
123
  else:
124
  logger.warning(
 
211
  "gpu_name": torch.cuda.get_device_name(0),
212
  "vram_used_mb": round(torch.cuda.memory_allocated(0) / 1024**2),
213
  "vram_total_mb": round(
214
+ torch.cuda.get_device_properties(0).total_memory / 1024**2
215
  ),
216
  }
217
  return {}