()
| 82 | print("-" * (terminal_width-5)) |
| 83 | |
| 84 | def list_avail_models(): |
| 85 | from .model import official |
| 86 | model_list = [] |
| 87 | for name in dir(official): |
| 88 | # if is a class |
| 89 | if isinstance(getattr(official, name), type): |
| 90 | model_list.append(name) |
| 91 | print('Available model definitions (example: "from sat import GLMModel"):') |
| 92 | print_aligned_string_list(model_list) |
| 93 | return model_list |
| 94 | |
| 95 | def list_avail_pretrained(): |
| 96 | from sat.resources.urls import MODEL_URLS |