From ad89bb14135bbfa3000c4d583d66f51cc041a7ff Mon Sep 17 00:00:00 2001 From: Chi_Liu Date: Fri, 2 Jun 2023 14:44:46 -0700 Subject: [PATCH] Add distilgpt2 to stablehlo in shark tank (#1481) --- tank/torch_model_list.csv | 1 + 1 file changed, 1 insertion(+) 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