.vscode/launch.json

{
    // Use IntelliSense to learn about possible attributes.
    // Hover to view descriptions of existing attributes.
    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
    "version": "0.2.0",
    "configurations": [
        {
            "name": "3d_grand_demo",
            "type": "python",
            "request": "launch",
            "module": "app",
            "justMyCode": false,
            "env": {
                "CUDA_VISIBLE_DEVICES": "1",
            },
            "args": []
        },
        {
            "name": "merge_lora_weights",
            "type": "python",
            "request": "launch",
            "module": "scripts.merge_lora_weights",
            "justMyCode": false,
            "env": {
                "CUDA_VISIBLE_DEVICES": "1",
            },
            "args": [
                "--model-base=/data/jianingy/3d_grand_demo_hf_space/checkpoints/llava-llama-2-7b-chat-lightning-preview",
                "--model-path=/data/jianingy/3d_grand_demo_hf_space/checkpoints/llava_lora_grounded_obj_ref_checkpoint-4896",
                "--save-model-path=/data/jianingy/3d_grand_demo_hf_space/checkpoints/merged_weights_grounded_obj_ref"
            ]
        }
    ]
}