Spaces:

Anonymous-sub
/

Rerender

Sleeping

Anonymous-sub commited on Jun 16, 2023

Commit

dc716b9

•

1 Parent(s): da6c999

Update flow/flow_utils.py

Files changed (1) hide show

flow/flow_utils.py CHANGED Viewed

@@ -18,7 +18,7 @@ repo_name = 'Anonymous-sub/Rerender'
 global_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 gmflow_path = huggingface_hub.hf_hub_download(
-    repo_name, 'models/gmflow_sintel-0c07dcb3.pth', local_dir='models')
 def coords_grid(b, h, w, homogeneous=False, device=None):

 global_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 gmflow_path = huggingface_hub.hf_hub_download(
+    repo_name, 'models/gmflow_sintel-0c07dcb3.pth', local_dir='./')
 def coords_grid(b, h, w, homogeneous=False, device=None):