Spaces:

xiaosu-zhu
/

McQuic

Runtime error

xiaosu-zhu commited on Mar 17, 2022

Commit

89d1722

1 Parent(s): 0c221a6

Auto deploy

Files changed (1) hide show

stCompressService.py CHANGED Viewed

@@ -21,7 +21,7 @@ MODELS_URL = "https://github.com/xiaosu-zhu/McQuic/releases/download/generic/qp_
 @st.experimental_singleton
 def loadModel(qp: int, local: pathlib.Path, device, mse: bool):
-    ckpt = torch.hub.load_state_dict_from_url(MODELS_URL, map_location=device)
     config = Config.deserialize(ckpt["config"])
     model = Compressor(**config.Model.Params).to(device)

 @st.experimental_singleton
 def loadModel(qp: int, local: pathlib.Path, device, mse: bool):
+    ckpt = torch.hub.load_state_dict_from_url(MODELS_URL, map_location=device, check_hash=True)
     config = Config.deserialize(ckpt["config"])
     model = Compressor(**config.Model.Params).to(device)