瀏覽代碼

get deepbooru to run with --precision-half

AUTOMATIC1111 1 年之前
父節點
當前提交
06fe174c74
共有 1 個文件被更改,包括 1 次插入1 次删除
  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
 
         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()
 
         probability_dict = {}