coreml-community/ControlNet-v1-1-Annotators-cpu
15
1# -*- coding: utf-8 -*-2# Copyright (c) Facebook, Inc. and its affiliates.3from dataclasses import dataclass4from typing import Optional5 6 7@dataclass8class ShapeSpec:9 """10 A simple structure that contains basic shape specification about a tensor.11 It is often used as the auxiliary inputs/outputs of models,12 to complement the lack of shape inference ability among pytorch modules.13 """14 15 channels: Optional[int] = None16 height: Optional[int] = None17 width: Optional[int] = None18 stride: Optional[int] = None19 