minmax performance on those tensor cores lol

inference_gradio.py

@@ -123,7 +123,7 @@ model.eval()
 if torch.cuda.is_available():
     model.cuda()
     if torch.cuda.get_device_capability()[0] >= 7: # tensor cores
-        model.half()
+        model.to(dtype=torch.float16, memory_format=torch.channels_last)
 
 with open("JTP_PILOT/tags.json", "r") as file:
     tags = json.load(file) # type: dict
@@ -139,7 +139,7 @@ def create_tags(image, threshold):
     if torch.cuda.is_available():
         tensor.cuda()
         if torch.cuda.get_device_capability()[0] >= 7:
-            tensor.half()
+            tensor.to(dtype=torch.float16, memory_format=torch.channels_last)
 
     with torch.no_grad():
         logits = model(tensor)