Merge pull request caikit#116 from gkumbhat/consolidate_tgis_finetuning

Consolidate tgis finetuning Signed-off-by: gkumbhat <[email protected]>
gkumbhat · Aug 8, 2023 · 84e4ee4 · 84e4ee4
2 parents 5088976 + 6907c9d
commit 84e4ee4
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