Skip to content

Commit

Permalink
fix: 🐞 temporarly ignore mypy errors for Literal/typer
Browse files Browse the repository at this point in the history
Signed-off-by: Onuralp SEZER <[email protected]>
  • Loading branch information
SkalskiP authored and onuralpszr committed Sep 24, 2024
1 parent c89b5ba commit 3042e50
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions maestro/trainer/models/florence_2/entrypoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,19 +138,19 @@ 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,
val_num_workers=val_num_workers,
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,
)
Expand Down

0 comments on commit 3042e50

Please sign in to comment.