From 9d31dbc7b69622b3ac3b966b463352a0eef0e622 Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Tue, 24 Sep 2024 15:15:02 +0200 Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9E=20temporarly=20ignore=20=20?= =?UTF-8?q?mypy=20errors=20for=20Literal/typer?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Onuralp SEZER --- maestro/trainer/models/florence_2/entrypoint.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/maestro/trainer/models/florence_2/entrypoint.py b/maestro/trainer/models/florence_2/entrypoint.py index f8ae79b..4787c69 100644 --- a/maestro/trainer/models/florence_2/entrypoint.py +++ b/maestro/trainer/models/florence_2/entrypoint.py @@ -138,9 +138,9 @@ def train( device=torch.device(device), cache_dir=cache_dir, epochs=epochs, - optimizer=optimizer, + optimizer=optimizer, # type: ignore lr=lr, - lr_scheduler=lr_scheduler, + lr_scheduler=lr_scheduler, # type: ignore batch_size=batch_size, val_batch_size=val_batch_size, num_workers=num_workers, @@ -148,9 +148,9 @@ def train( lora_r=lora_r, lora_alpha=lora_alpha, lora_dropout=lora_dropout, - bias=bias, + bias=bias, # type: ignore use_rslora=use_rslora, - init_lora_weights=init_lora_weights, + init_lora_weights=init_lora_weights, # type: ignore output_dir=output_dir, metrics=metric_objects, )