Spaces:

xiaosu-zhu
/

McQuic

Runtime error

xiaosu-zhu commited on Mar 28, 2022

Commit

c30bbb7

1 Parent(s): 277a2e9

Auto deploy

Files changed (1) hide show

stCompressService.py CHANGED Viewed

@@ -30,7 +30,7 @@ def loadModel(device):
     config = Config.deserialize(ckpt["config"])
     model = Compressor(**config.Model.Params).to(device)
-    model.QuantizationParameter = str(local) if local is not None else "qp_2_msssim"
     model.load_state_dict(ckpt["model"])
     return model

     config = Config.deserialize(ckpt["config"])
     model = Compressor(**config.Model.Params).to(device)
+    model.QuantizationParameter = "qp_2_msssim"
     model.load_state_dict(ckpt["model"])
     return model