(model_name)
| 73 | |
| 74 | |
| 75 | def download(model_name): |
| 76 | url = MODELS_MAP[model_name]["url"] |
| 77 | r = requests.get(url, stream=True) |
| 78 | |
| 79 | local_filename = f"./{model_name}.ckpt" |
| 80 | with open(local_filename, "wb") as fp: |
| 81 | for chunk in r.iter_content(chunk_size=8192): |
| 82 | fp.write(chunk) |
| 83 | |
| 84 | return local_filename |
| 85 | |
| 86 | |
| 87 | DOWN_NUM_TO_LAYER = { |