float16 rm

app.py

@@ -31,7 +31,7 @@ model = model_factory.load_model('vitb_8x8patch_3frames')#.to(device)
 model.requires_grad_(False)
 model.eval()
 
-model = model.to(torch.float16)
+model = model#.to(torch.float16)
 
 
 import matplotlib.pyplot as plt
@@ -273,7 +273,7 @@ with gr.Blocks() as demo:
             img = img[None]
 
             # reshape image to [B, C, T, H, W], C = 3, T = 3 (3-frame model), H = W = 224
-            x = img[:, :, None].expand(-1, -1, 3, -1, -1).to(torch.float16)
+            x = img[:, :, None].expand(-1, -1, 3, -1, -1)#.to(torch.float16)
 
             # Imagenet-normalize the inputs (standardization)