File size: 572 Bytes
e15b26d
d9bccac
e15b26d
 
2dd31d8
d9bccac
 
 
2495ef6
 
d9bccac
1
2
3
4
5
6
7
8
9
10
11
import os
import huggingface_hub
os.system("git clone https://github.com/etched-ai/open-oasis.git")
os.system("cd open-oasis")
os.system("pip install -r open-oasis/requirements.txt")
#os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt") # DiT checkpoint
#os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt") 
from huggingface_hub import hf_hub_download
oas = hf_hub_download(repo_id="Etched/oasis-500m", token=None, filename="oasis500m.pt")
vit = hf_hub_download(repo_id="Etched/oasis-500m", token=None filename="vit-l-20.pt")
print(oas,vit)