Upload folder using huggingface_hub

README.md

@@ -12,11 +12,6 @@ tags:
   - lora
 ---
 
-# LuxDiT
-
-This is the model checkpoint for [LuxDiT](https://github.com/nv-tlabs/LuxDiT): Lighting Estimation with Video Diffusion Transformer. It is finetuned on image data and includes a LoRA adapter for real scenes.
-
-![](https://research.nvidia.com/labs/toronto-ai/LuxDiT/assets/teaser.png)
 
 ## Model description
 
@@ -29,6 +24,8 @@ LuxDiT is a generative lighting estimation model that predicts high-quality HDR
 
 **Use case:** LuxDiT supports studies and prototyping in video lighting estimation. This release is an open-source implementation of our research paper, intended for AI research, development, and benchmarking for lighting estimation research.
 
+![](https://research.nvidia.com/labs/toronto-ai/LuxDiT/assets/teaser.png)
+
 ### Model architecture
 
 - **Architecture type:** Transformer (based on [CogVideoX](https://github.com/THUDM/CogVideoX))
@@ -49,7 +46,7 @@ LuxDiT is a generative lighting estimation model that predicts high-quality HDR
 From the [LuxDiT repository](https://github.com/nv-tlabs/LuxDiT) root:
 
 ```bash
-python download_weights.py --repo_id <HF_ORG>/LuxDiT
+python download_weights.py --repo_id nvidia/LuxDiT
 ```
 
 This saves the checkpoint to `checkpoints/LuxDiT` by default. Use `--local_dir` to override.
@@ -196,8 +193,8 @@ python hdr_merger.py \
 
 | Checkpoint   | Description |
 |-------------|-------------|
-| [luxdit_image](https://huggingface.co/nvidia/LuxDiT/luxdit_image) | Image-finetuned, with LoRA for real scenes |
-| [luxdit_video](https://huggingface.co/nvidia/LuxDiT/luxdit_video) | Video-finetuned, with LoRA for real scenes |
+| [luxdit_image](https://huggingface.co/nvidia/LuxDiT/tree/main/luxdit_image) | Image-finetuned, with LoRA for real scenes |
+| [luxdit_video](https://huggingface.co/nvidia/LuxDiT/tree/main/luxdit_video) | Video-finetuned, with LoRA for real scenes |
 
 For video inputs and object scans, use **luxdit_video** instead.