mirror of
https://github.com/facebookresearch/blt.git
synced 2025-09-01 18:19:06 +00:00
96d51b59d2
Summary: Add code to download weights and demo code for running model. Weights at: - https://huggingface.co/collections/facebook/blt-6801263d4ac1704702a192a6 - https://huggingface.co/facebook/blt - https://huggingface.co/facebook/blt-1b - https://huggingface.co/facebook/blt-7b Test Plan:
15 lines
353 B
Python
15 lines
353 B
Python
import os
|
|
|
|
import typer
|
|
from huggingface_hub import snapshot_download
|
|
|
|
|
|
def main(models: list[str] = ["blt-1b", "blt-7b"]):
|
|
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}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
typer.run(main)
|