storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-22 08:05:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Switching to Full FineTuning

Browse Source
This commit is contained in:
CrossProduct
2025-01-09 00:27:15 +00:00
parent 72ec207708
commit d0ee9c3101
8 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+1 -1
View File
@@ -395,7 +395,7 @@ class Trainer:
]
)
if self.args.training_type == "sft":
if self.args.training_type == "full-finetune":
logger.info("Full Fine Tuning Enabled")
self._enable_grad_for_components(components=[self.transformer])
else: