summaryrefslogtreecommitdiff
path: root/finetuning.py
diff options
context:
space:
mode:
Diffstat (limited to 'finetuning.py')
-rwxr-xr-xfinetuning.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/finetuning.py b/finetuning.py
index 62f506c..a0ad1fe 100755
--- a/finetuning.py
+++ b/finetuning.py
@@ -80,7 +80,7 @@ def main():
parser.add_argument("--max-length", default=512, type=int)
parser.add_argument("--lora-config", choices=["S", "M", "L"], default="M")
args = parser.parse_args()
-
+
if args.bnb_4bit:
bnb_config = BitsAndBytesConfig(
load_in_4bit=True,