diff --git a/tank/torch_model_list.csv b/tank/torch_model_list.csv index 3298ae6a..c1f1b8a4 100644 --- a/tank/torch_model_list.csv +++ b/tank/torch_model_list.csv @@ -24,3 +24,4 @@ bert-large-uncased,True,hf,True,linalg,False,330M,"nlp;bert-variant;transformer- bert-base-uncased,True,hf,False,stablehlo,False,109M,"nlp;bert-variant;transformer-encoder","12 layers; 768 hidden; 12 attention heads" gpt2,True,hf_causallm,False,stablehlo,True,125M,"nlp;transformer-encoder","-" facebook/opt-125m,True,hf,False,stablehlo,True,125M,"nlp;transformer-encoder","-" +distilgpt2,True,hf,False,stablehlo,True,88M,"nlp;transformer-encoder","-" \ No newline at end of file