LLama3 download fix (#14508)

JohnSnowLabs · Jan 29, 2025 · 7a9f845 · 7a9f845
1 parent 8deb7da
commit 7a9f845
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala b/src/main/scala/com/johnsnowlabs/nlp/pretrained/ResourceDownloader.scala
@@ -682,6 +682,7 @@ object PythonResourceDownloader {
     "MPNetForSequenceClassification" -> MPNetForSequenceClassification,
     "MPNetForQuestionAnswering" -> MPNetForQuestionAnswering,
     "LLAMA2Transformer" -> LLAMA2Transformer,
+    "LLAMA3Transformer" -> LLAMA3Transformer,
     "M2M100Transformer" -> M2M100Transformer,
     "UAEEmbeddings" -> UAEEmbeddings,
     "AutoGGUFModel" -> AutoGGUFModel,
@@ -691,7 +692,7 @@ object PythonResourceDownloader {
     "CamemBertForZeroShotClassification" -> CamemBertForZeroShotClassification,
     "BertForMultipleChoice" -> BertForMultipleChoice,
     "PromptAssembler" -> PromptAssembler,
-    "CPMTransformer"-> CPMTransformer,
+    "CPMTransformer" -> CPMTransformer,
     "NomicEmbeddings" -> NomicEmbeddings,
     "NLLBTransformer" -> NLLBTransformer,
     "Phi3Transformer" -> Phi3Transformer,