|
@@ -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 = {}
|