revert another change

togethercomputer · Aug 21, 2024 · d62638c · d62638c
1 parent 8faa645
commit d62638c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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,