Fix DarkStage device kwargs

Author: rwightman

timm/models/cspnet.py

@@ -500,6 +500,8 @@ def __init__(
  avg_down: bool = False,
  block_fn: Type[nn.Module] = BottleneckBlock,
  block_dpr: Optional[List[float]] = None,
+ device=None,
+ dtype=None,
  **block_kwargs,
  ):
  dd = {'device': device, 'dtype': dtype}