KeyError when running infer.sh

[sselgrad23]

Hi! You have some impressive results, and I wanted to evaluate your method. After having run the dataset extract script for the kitti dataset, I decided to run the infer.sh script. Unfortunately, I seem to be running into the following errors:

+ gpu_id=0
+ process_length=110
+ saved_root=/cluster/scratch/bla/eval_dataset
+ saved_dataset_folder=results_kitti/KITTI
+ overlap=1
+ dataset=kitti
+ CUDA_VISIBLE_DEVICES=0 PYTHONPATH=. python run.py --video-path /cluster/scratch/bla/eval_dataset/KITTI/2011_10_03_drive_0047_sync_rgb_left.mp4 --save-folder /cluster/scratch/bla/eval_dataset/results_kitti/KITTI --process-length 110 --dataset kitti --overlap 1
/cluster/scratch/bla/DepthCrafter/venv/depthcrafter/lib/python3.11/site-packages/diffusers/models/transformers/transformer_2d.py:34: FutureWarning: `Transformer2DModelOutput` is deprecated and will be removed in version 1.0.0. Importing `Transformer2DModelOutput` from `diffusers.models.transformer_2d` is deprecated and this will be removed in a future version. Please use `from diffusers.models.modeling_outputs import Transformer2DModelOutput`, instead.
  deprecate("Transformer2DModelOutput", "1.0.0", deprecation_message)
Loading pipeline components...: 100%|██████████| 5/5 [00:00<00:00,  5.81it/s]
Traceback (most recent call last):
  File "/cluster/scratch/bla/DepthCrafter/run.py", line 209, in <module>
    Fire(main)
  File "/cluster/scratch/bla/DepthCrafter/venv/depthcrafter/lib/python3.11/site-packages/fire/core.py", line 143, in Fire
    component_trace = _Fire(component, args, parsed_flag_args, context, name)
                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/cluster/scratch/bla/DepthCrafter/venv/depthcrafter/lib/python3.11/site-packages/fire/core.py", line 477, in _Fire
    component, remaining_args = _CallAndUpdateTrace(
                                ^^^^^^^^^^^^^^^^^^^^
  File "/cluster/scratch/bla/DepthCrafter/venv/depthcrafter/lib/python3.11/site-packages/fire/core.py", line 693, in _CallAndUpdateTrace
    component = fn(*varargs, **kwargs)
                ^^^^^^^^^^^^^^^^^^^^^^
  File "/cluster/scratch/bla/DepthCrafter/run.py", line 183, in main
    depthcrafter_demo.infer(
  File "/cluster/scratch/bla/DepthCrafter/run.py", line 72, in infer
    frames, target_fps = read_video_frames(
                         ^^^^^^^^^^^^^^^^^^
  File "/cluster/scratch/bla/DepthCrafter/depthcrafter/utils.py", line 32, in read_video_frames
    height = dataset_res_dict[dataset][0]
             ~~~~~~~~~~~~~~~~^^^^^^^^^
KeyError: 'kitti'

If you know how to prevent these errors, I would be very grateful if you could inform me how to do so.