Browse Source

Log the pytorch version.

pull/3493/merge
comfyanonymous 6 months ago
parent
commit
09e069ae6c
  1. 5
      comfy/model_management.py

5
comfy/model_management.py

@ -120,6 +120,11 @@ total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
total_ram = psutil.virtual_memory().total / (1024 * 1024)
logging.info("Total VRAM {:0.0f} MB, total RAM {:0.0f} MB".format(total_vram, total_ram))
try:
logging.info("pytorch version: {}".format(torch.version.__version__))
except:
pass
try:
OOM_EXCEPTION = torch.cuda.OutOfMemoryError
except:

Loading…
Cancel
Save