fix glide import error

This commit is contained in:
anton-l 2022-06-15 15:46:20 +02:00
parent 84bd65bced
commit 8e5945a76e
2 changed files with 6 additions and 1 deletions

View File

@ -1,6 +1,10 @@
from .pipeline_bddm import BDDM
from .pipeline_ddim import DDIM
from .pipeline_ddpm import DDPM
from .pipeline_glide import GLIDE
try:
from .pipeline_glide import GLIDE
except ImportError:
class GLIDE:
pass
from .pipeline_latent_diffusion import LatentDiffusion
from .pipeline_pndm import PNDM

View File

@ -15,6 +15,7 @@
""" PyTorch CLIP model."""
import math
import logging
from dataclasses import dataclass
from typing import Any, Optional, Tuple, Union