Commit eb5267f3 authored by anton-l's avatar anton-l
Browse files

Style quickfix

parent db5fa430
...@@ -337,7 +337,9 @@ class ModelMixin(torch.nn.Module): ...@@ -337,7 +337,9 @@ class ModelMixin(torch.nn.Module):
) )
if torch_dtype is not None and not isinstance(torch_dtype, torch.dtype): if torch_dtype is not None and not isinstance(torch_dtype, torch.dtype):
raise ValueError(f"{torch_dtype} needs to be of type `torch.dtype`, e.g. `torch.float16`, but is {type(torch_dtype)}.") raise ValueError(
f"{torch_dtype} needs to be of type `torch.dtype`, e.g. `torch.float16`, but is {type(torch_dtype)}."
)
elif torch_dtype is not None: elif torch_dtype is not None:
model = model.to(torch_dtype) model = model.to(torch_dtype)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment