@ -34,6 +34,9 @@ if args.directml is not None:
try:
import torch
if directml_enabled:
total_vram = 4097 #TODO
else:
import intel_extension_for_pytorch as ipex
if torch.xpu.is_available():