From d62638c0cd252e5ae5b951db09d0203019eb90d4 Mon Sep 17 00:00:00 2001
From: Artem Chumachenko <artek.chumak@gmail.com>
Date: Wed, 21 Aug 2024 08:59:50 +0200
Subject: [PATCH] revert another change

---
 src/together/resources/finetune.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/together/resources/finetune.py b/src/together/resources/finetune.py
index f9496522..e2cdcaa2 100644
--- a/src/together/resources/finetune.py
+++ b/src/together/resources/finetune.py
@@ -72,15 +72,14 @@ def create(
             client=self._client,
         )
 
+        training_type: TrainingType = FullTrainingType()
         if lora:
-            training_type: TrainingType = LoRATrainingType(
+            training_type = LoRATrainingType(
                 lora_r=lora_r,
                 lora_alpha=lora_alpha,
                 lora_dropout=lora_dropout,
                 lora_trainable_modules=lora_trainable_modules,
             )
-        else:
-            training_type = FullTrainingType()
 
         parameter_payload = FinetuneRequest(
             model=model,