11 lines
429 B
Python
11 lines
429 B
Python
from ...utils import OptionalDependencyNotAvailable, is_torch_available, is_transformers_available
|
|
|
|
|
|
try:
|
|
if not (is_transformers_available() and is_torch_available()):
|
|
raise OptionalDependencyNotAvailable()
|
|
except OptionalDependencyNotAvailable:
|
|
from ...utils.dummy_torch_and_transformers_objects import *
|
|
else:
|
|
from .pipeline_vq_diffusion import LearnedClassifierFreeSamplingEmbeddings, VQDiffusionPipeline
|