Spaces:

stevengrove
/

YOLO-World

Runtime error

stevengrove commited on Jan 31, 2024

Commit

e789019

verified ·

1 Parent(s): fa52fdd

Update tools/demo.py

Files changed (1) hide show

tools/demo.py CHANGED Viewed

@@ -116,6 +116,7 @@ def export_model(runner,
     # dry run
     deploy_model(fake_input)
     save_onnx_path = os.path.join(
         args.work_dir,
         os.path.basename(args.checkpoint).replace('pth', 'onnx'))

     # dry run
     deploy_model(fake_input)
+    os.makedirs(args.work_dir, exist_ok=True)
     save_onnx_path = os.path.join(
         args.work_dir,
         os.path.basename(args.checkpoint).replace('pth', 'onnx'))