black

vince62s · Nov 8, 2023 · e99e7fc · e99e7fc
1 parent 589d4f1
commit e99e7fc
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/onmt/models/model.py b/onmt/models/model.py
@@ -140,7 +140,9 @@ def load_state_dict(
  and checkpoint["generator"] is not None
  and param_name in checkpoint["generator"].keys()
  ):
- keyname = name + "." + param_name if "linear" in name else param_name
+ keyname = (
+ name + "." + param_name if "linear" in name else param_name
+ )
  param.data = checkpoint["generator"][keyname]
  del checkpoint["generator"][keyname]
  elif strict and "lora" not in param_name: