mirror of
https://github.com/vladmandic/automatic
synced 2026-09-18 16:54:33 +02:00
Autodetect IPEX
This commit is contained in:
@@ -101,7 +101,7 @@ def setup_model(dirname):
|
||||
output = self.net(cropped_face_t, w=w if w is not None else shared.opts.code_former_weight, adain=True)[0]
|
||||
restored_face = tensor2img(output, rgb2bgr=True, min_max=(-1, 1))
|
||||
del output
|
||||
if cmd_opts.use_ipex:
|
||||
if devices.backend == 'ipex':
|
||||
torch.xpu.empty_cache()
|
||||
else:
|
||||
torch.cuda.empty_cache()
|
||||
|
||||
Reference in New Issue
Block a user