Merge pull request caikit#116 from gkumbhat/consolidate_tgis_finetuning

Consolidate tgis finetuning
Maxusmusti · Aug 8, 2023 · 8ece6e3 · 8ece6e3
2 parents 4eb05cf + 0f71982
commit 8ece6e3
Show file tree

Hide file tree

Showing 12 changed files with 655 additions and 553 deletions.
diff --git a/caikit_nlp/modules/text_generation/__init__.py b/caikit_nlp/modules/text_generation/__init__.py
@@ -13,7 +13,6 @@
 # limitations under the License.
 
 # Local
-from .fine_tuning import FineTuning
 from .peft_prompt_tuning import PeftPromptTuning
 from .peft_tgis_remote import PeftPromptTuningTGIS
 from .text_generation_local import TextGeneration

diff --git a/caikit_nlp/modules/text_generation/fine_tuning.py b/caikit_nlp/modules/text_generation/fine_tuning.py