{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "8cfa8aa6-d276-417e-8903-10887f6aebca", "metadata": {}, "outputs": [], "source": [ "import huggingface_hub as hub" ] }, { "cell_type": "code", "execution_count": 34, "id": "31c61098-c6ee-4466-a2e5-60337589367a", "metadata": {}, "outputs": [], "source": [ "CEBAB = 'CEBaB'" ] }, { "cell_type": "code", "execution_count": 2, "id": "f9f929bc-5e76-445a-8e7d-43d4656c9583", "metadata": {}, "outputs": [], "source": [ "cebab_models = hub.list_models(filter=hub.ModelFilter(author=CEBAB))" ] }, { "cell_type": "code", "execution_count": 4, "id": "0cd3bf15-58a4-4485-ac0b-b941748c6d12", "metadata": {}, "outputs": [], "source": [ "causalm_models = sorted([m.id for m in cebab_models if 'causalm' in m.id])" ] }, { "cell_type": "code", "execution_count": 25, "id": "c608e3ba-52fa-4fbf-aa07-1158938a8596", "metadata": {}, "outputs": [], "source": [ "bert_causalm = [m for m in causalm_models if 'bert-base-uncased' in m and '__' in m]\n", "roberta_causalm = [m for m in causalm_models if 'roberta-base' in m]\n", "lstm_causalm = [m for m in causalm_models if 'lstm' in m]\n", "gpt2_causalm = [m for m in causalm_models if 'gpt2' in m]" ] }, { "cell_type": "code", "execution_count": 36, "id": "67ce63fc-87d7-4024-9b6e-9430b7be54c7", "metadata": {}, "outputs": [ { "data": { "application/vnd.jupyter.widget-view+json": { "model_id": "a02948a164c74b80960a27911cd4a53a", "version_major": 2, "version_minor": 0 }, "text/plain": [ "VBox(children=(HTML(value='