madebyollin

# ------------------------------------------------------------------
# EDIT: I eventually found a faster way to run SD on macOS, via MPSGraph (~0.8s / step on M1 Pro):
#   https://github.com/madebyollin/maple-diffusion
# The original CoreML-related code & discussion is preserved below :)
# ------------------------------------------------------------------

# you too can run stable diffusion on the apple silicon GPU (no ANE sadly)
#
# quick test portraits (each took 50 steps x 2s / step ~= 100s on my M1 Pro):
# * https://i.imgur.com/5ywISvm.png

rwightman

from typing import Callable, Union
from dataclasses import dataclass

import timm
import torch.nn as nn
from effdet.efficientdet import BiFpn
from effdet.config import fpn_config

from omegaconf import DictConfig

hamelsmu