Browse Source

get deepbooru to run with --precision-half

AUTOMATIC1111 1 năm trước cách đây
mục cha
commit
06fe174c74
1 tập tin đã thay đổi với 1 bổ sung1 xóa
  1. 1 1
      modules/deepbooru.py

+ 1 - 1
modules/deepbooru.py

@@ -57,7 +57,7 @@ class DeepDanbooru:
         a = np.expand_dims(np.array(pic, dtype=np.float32), 0) / 255
         a = np.expand_dims(np.array(pic, dtype=np.float32), 0) / 255
 
 
         with torch.no_grad(), devices.autocast():
         with torch.no_grad(), devices.autocast():
-            x = torch.from_numpy(a).to(devices.device)
+            x = torch.from_numpy(a).to(devices.device, devices.dtype)
             y = self.model(x)[0].detach().cpu().numpy()
             y = self.model(x)[0].detach().cpu().numpy()
 
 
         probability_dict = {}
         probability_dict = {}