CoolFace
Apppublic

coreml-community/ControlNet-v1-1-Annotators-cpu

sourceHugging Facemitupdated 2y agoView on Hugging Face
15likes
shape_spec.py19 linesDownload Raw Back to layers
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