vikarti.anatra/blt
1
0
Fork 0
mirror of https://github.com/facebookresearch/blt.git synced 2025-09-15 08:39:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix demo.py (#116)

Browse source 
Summary:

Test Plan:
This commit is contained in:
Pedro Rodriguez 2025-05-22 12:15:57 -07:00 • committed by GitHub
parent d286bbe415
commit 3dd5e9fb62
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

5
download_blt_weights.py
View file

@ -4,11 +4,10 @@ import typer
from huggingface_hub import snapshot_download
def main(models: list[str] = ["blt-1b", "blt-7b"]):
def main():
if not os.path.exists("hf-weights"):
os.makedirs("hf-weights")
for model in models:
snapshot_download(f"facebook/{model}", local_dir=f"hf-weights/{model}")
snapshot_download(f"facebook/blt", local_dir=f"hf-weights")
if __name__ == "__main__":