File size: 1,153 Bytes
e15b26d
9a40f8a
9c3d12f
ec07bb5
fbe69f5
 
fc7a6ef
ea0a988
 
fbe69f5
b2615f7
7b282e8
b2615f7
da60df0
7b282e8
b2615f7
9c3d12f
e15b26d
2dd31d8
10f43d5
 
 
 
 
 
9c3d12f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import os
import gradio as gr
#import huggingface_hub
#from huggingface_hub import snapshot_download
#import open_oasis_master
#from open_oasis_master import generate
os.system("huggingface-cli login --token $HF_TOKEN --add-to-git-credential")
os.system("huggingface-cli download Etched/oasis-500m oasis500m.pt --local-dir ./") # DiT checkpoint
os.system("huggingface-cli download Etched/oasis-500m vit-l-20.pt --local-dir ./") 
os.system("cd open_oasis_master && python generate.py")

'''with gr.Blocks() as app:
    vid = gr.Video()
    app.load(generate.run_mod(),None,vid)
app.launch()'''

'''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 login")
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", filename="oasis500m.pt")
#vit = hf_hub_download(repo_id="Etched/oasis-500m", filename="vit-l-20.pt")
#print(oas,vit)'''