Spaces:

haakohu
/

deep_privacy2

Runtime error

haakohu commited on Nov 17, 2022

Commit

a1cdc0f

1 Parent(s): 548d634

fix

Files changed (1) hide show

dp2/infer.py CHANGED Viewed

@@ -27,7 +27,7 @@ def build_trained_generator(cfg, map_location=None):
     else:
         ckpt_mapper = None
     if "model_url" in cfg.common:
-        ckpt = tops.load_file_or_url(cfg.common.model_url, md5sum=cfg.common.model_md5sum)
         load_generator_state(ckpt, G, ckpt_mapper)
         return G
     try:

     else:
         ckpt_mapper = None
     if "model_url" in cfg.common:
+        ckpt = tops.load_file_or_url(cfg.common.model_url, md5sum=cfg.common.model_md5sum, map_location=torch.device("cpu"))
         load_generator_state(ckpt, G, ckpt_mapper)
         return G
     try: