Add config to dataloader alignment

This commit is contained in:
henryruhs
2025-03-03 09:39:56 +01:00
parent cc9a0ba83e
commit 1ab57b1d3f
+1 -2
View File
@@ -1,6 +1,5 @@
from typing import Any, Dict, Tuple, TypeAlias
from typing import Any, Dict, Tuple, TypeAlias, Type, Literal
from jinja2.nodes import Literal
from torch import Tensor
from torch.nn import Module