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