()
| 93 | return model_list |
| 94 | |
| 95 | def list_avail_pretrained(): |
| 96 | from sat.resources.urls import MODEL_URLS |
| 97 | # iterate over all pretrained models into a list |
| 98 | model_list = [] |
| 99 | for model_name, model_url in MODEL_URLS.items(): |
| 100 | model_list.append(model_name) |
| 101 | print('Available pretrained models (example: sat.AutoModel.from_pretrained("roberta-base")):') |
| 102 | print_aligned_string_list(model_list) |
| 103 | return model_list |
| 104 | |
| 105 | import logging |
| 106 | import torch |