Sfoglia il codice sorgente

get deepbooru to run with --precision-half

AUTOMATIC1111 1 anno fa
parent
commit
06fe174c74
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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 = {}