Merge pull request #277 from DashAISoftware/project/generative-plugin-llm-fix

Project/generative plugin llm fix

Author: cristian-tamblay

DashAI/back/initial_components.py

@@ -76,7 +76,6 @@
     DecisionTreeClassifier,
     DistilBertTransformer,
     DummyClassifier,
-    GemmaModel,
     GradientBoostingR,
     HistGradientBoostingClassifier,
     KNeighborsClassifier,
@@ -145,7 +144,6 @@ def get_initial_components():
         GradientBoostingR,
         HistGradientBoostingClassifier,
         KNeighborsClassifier,
-        GemmaModel,
         QwenModel,
         StableDiffusionV2Model,
         StableDiffusionV3Model,

DashAI/back/models/__init__.py

@@ -2,7 +2,6 @@
 from DashAI.back.models.base_generative_model import BaseGenerativeModel
 from DashAI.back.models.base_model import BaseModel
 from DashAI.back.models.hugging_face.distilbert_transformer import DistilBertTransformer
-from DashAI.back.models.hugging_face.gemma_model import GemmaModel
 from DashAI.back.models.hugging_face.opus_mt_en_es_transformer import (
     OpusMtEnESTransformer,
 )