diff --git a/evaluation/cosmos3/generator/rbench/run_with_cosmos_framework.ipynb b/evaluation/cosmos3/generator/rbench/run_with_cosmos_framework.ipynb index faa6a93f..016e8584 100644 --- a/evaluation/cosmos3/generator/rbench/run_with_cosmos_framework.ipynb +++ b/evaluation/cosmos3/generator/rbench/run_with_cosmos_framework.ipynb @@ -99,7 +99,7 @@ "COSMOS3_NUM_GPUS = os.environ.get(\"COSMOS3_NUM_GPUS\", \"4\")\n", "CUDA_VISIBLE_DEVICES = os.environ.get(\"CUDA_VISIBLE_DEVICES\", \"0,1,2,3\")\n", "\n", - "RBENCH_NOTEBOOK_ROOT = COSMOS_ROOT / \"evaluation\" / \"rbench\"\n", + "RBENCH_NOTEBOOK_ROOT = COSMOS_ROOT / \"evaluation\" / \"cosmos3\" / \"generator\" / \"rbench\"\n", "RBENCH_ASSETS = RBENCH_NOTEBOOK_ROOT / \"assets\"\n", "RBENCH_PROMPTS_DIR = RBENCH_ASSETS / \"prompts\"\n", "RBENCH_HF_URL = os.environ.get(\"RBENCH_HF_URL\", \"https://huggingface.co/datasets/DAGroup-PKU/RBench\")\n",