mark_dtype

app.py

@@ -15,7 +15,8 @@ ckpt_path = hf_hub_download(repo_id=ORIANY_V2, filename=REMOTE_CKPT_PATH, repo_t
 print(ckpt_path)
 
 
-mark_dtype = torch.bfloat16 if torch.cuda.get_device_capability()[0] >= 8 else torch.float16
+# mark_dtype = torch.bfloat16 if torch.cuda.get_device_capability()[0] >= 8 else torch.float16
+mark_dtype = torch.float16
 # device = 'cuda:0'
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')