update readme

README.md

@@ -4,23 +4,28 @@
 
 ![](./media/thumb.png)
 
-Oasis is an interactive world model developed by [Decart](https://www.decart.ai/) and [Etched](https://www.etched.com/). Based on diffusion transformers, Oasis takes in user keyboard input and generates gameplay in an autoregressive manner. Here we release the weights for Oasis 500M, a downscaled version of the model, along with inference code for action-conditional frame generation.
+Oasis is an interactive world model developed by [Decart](https://www.decart.ai/) and [Etched](https://www.etched.com/). Based on diffusion transformers, Oasis takes in user keyboard input and generates gameplay in an autoregressive manner. We release the weights for Oasis 500M, a downscaled version of the model, along with inference code for action-conditional frame generation.
 
 For more details, see our [joint blog post](https://oasis-model.github.io/) to learn more.
 
-## Setup
 
+## Setup
 ```
-git lfs install
-git clone https://huggingface.co/Etched/oasis-500m
-cd oasis-500m
+git clone https://github.com/etched-ai/open-oasis.git
+cd open-oasis
 pip install -r requirements.txt
 ```
 
+## Download the model weights
+```
+huggingface-cli login
+huggingface-cli download Etched/oasis-500m oasis500m.pt # DiT checkpoint
+huggingface-cli download Etched/oasis-500m vit-l-20.pt  # ViT VAE checkpoint
+```
+
 ## Basic Usage
 We include a basic inference script that loads a prompt frame from a video and generates additional frames conditioned on actions.
 ```
-cd oasis-500m
 python inference.py
 ```
 The resulting video will be saved to `video.mp4`.

requirements.txt

@@ -0,0 +1,31 @@
+av==13.1.0
+certifi==2024.8.30
+charset-normalizer==3.4.0
+diffusers==0.31.0
+einops==0.8.0
+filelock==3.13.1
+fsspec==2024.2.0
+huggingface-hub==0.26.2
+idna==3.10
+importlib_metadata==8.5.0
+Jinja2==3.1.3
+MarkupSafe==2.1.5
+mpmath==1.3.0
+networkx==3.2.1
+numpy==1.26.3
+packaging==24.1
+pillow==10.2.0
+PyYAML==6.0.2
+regex==2024.9.11
+requests==2.32.3
+safetensors==0.4.5
+sympy==1.13.1
+timm==1.0.11
+torch==2.5.1
+torchaudio==2.5.1
+torchvision==0.20.1
+tqdm==4.66.6
+triton==3.1.0
+typing_extensions==4.9.0
+urllib3==2.2.3
+zipp==3.20.2