Fixed the error related to CausalLM

chengchingwen · Dec 12, 2024 · af584b3 · af584b3
1 parent e21cc89
commit af584b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/huggingface/load.jl b/test/huggingface/load.jl
@@ -10,7 +10,7 @@
         :roberta => :[
             Model, ForMaskedLM, ForCausalLM, ForSequenceClassification, ForTokenClassification, ForQuestionAnswering,
         ].args,
-        :distilbert => :[Model, ForMaskedLM, ForCausalLM].args,
+        :distilbert => :[Model, ForMaskedLM].args,
         :gpt2 => [:Model, :LMHeadModel],
         :t5 => [:Model, :ForConditionalGeneration],
         :gpt_neo => [:Model, :ForCausalLM],