From a73716bf89cdfb8cc09c7e77d723f134104870a6 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 13 Aug 2024 23:35:56 +0300 Subject: [PATCH] Add meta to device check --- modules/sd_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 3f98e7d64..a842c5f2e 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -764,7 +764,7 @@ def set_diffuser_offload(sd_model, op: str = 'model'): sd_model.has_accelerate = True def normalize_device(device): - if torch.device(device).type in {"cpu", "mps"}: + if torch.device(device).type in {"cpu", "mps", "meta"}: return torch.device(device) if torch.device(device).index is None: return torch.device(str(device) + ":0")