Spaces:
Running
on
A10G
Running
on
A10G
Anonymous-sub
commited on
Commit
•
dc716b9
1
Parent(s):
da6c999
Update flow/flow_utils.py
Browse files- flow/flow_utils.py +1 -1
flow/flow_utils.py
CHANGED
@@ -18,7 +18,7 @@ repo_name = 'Anonymous-sub/Rerender'
|
|
18 |
|
19 |
global_device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
20 |
gmflow_path = huggingface_hub.hf_hub_download(
|
21 |
-
repo_name, 'models/gmflow_sintel-0c07dcb3.pth', local_dir='
|
22 |
|
23 |
|
24 |
def coords_grid(b, h, w, homogeneous=False, device=None):
|
|
|
18 |
|
19 |
global_device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
20 |
gmflow_path = huggingface_hub.hf_hub_download(
|
21 |
+
repo_name, 'models/gmflow_sintel-0c07dcb3.pth', local_dir='./')
|
22 |
|
23 |
|
24 |
def coords_grid(b, h, w, homogeneous=False, device=None):
|