File size: 823 Bytes
7b34e37 a4b0060 7b34e37 6fb939f 7b34e37 0461bfe 7b34e37 f901672 7b34e37 6fb939f 7b34e37 6fb939f 7b34e37 |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 |
"""
A script that is run when the server starts.
"""
from download_repo import download_repository
def disable_progress_bar():
"""
Disables the progress bar when downloading models.
"""
import transformers
transformers.logging.disable_progress_bar()
def download_useful_models():
"""
Downloads the models that are useful for this project.
So that the user doesn't have to wait for the models to download when they first use the app.
"""
print("Downloading useful models...")
useful_models = (
"facebook/opt-125m",
"facebook/opt-iml-max-30b",
)
for model_name in useful_models:
download_repository(
model_name,
)
def main():
disable_progress_bar()
download_useful_models()
if __name__ == "__main__":
main()
|