🐛 Fix clip device (#2560)

Mikubill · Jan 23, 2024 · f1f9493 · f1f9493
1 parent fc3b521
commit f1f9493
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/annotator/clipvision/__init__.py b/annotator/clipvision/__init__.py
@@ -115,6 +115,7 @@ def __init__(self, config, low_vram: bool):
         sd = torch.load(file_path, map_location=self.device)
         self.model.load_state_dict(sd, strict=False)
         del sd
+        self.model.to(self.device)
         self.model.eval()
 
     def unload_model(self):