SonicDiffusionClean / test_imports.py
alpercagann's picture
Add torch to requirements
dc72d06
import sys
print("Python version:", sys.version)
print("Python path:", sys.path)
try:
import diffusers
print("Diffusers version:", diffusers.__version__)
# Try importing specific classes from diffusers
from diffusers.configuration_utils import FrozenDict
print("Successfully imported FrozenDict")
from diffusers.loaders import IPAdapterMixin, LoraLoaderMixin, TextualInversionLoaderMixin
print("Successfully imported mixins")
from diffusers.models import AutoencoderKL, UNet2DConditionModel
print("Successfully imported models")
# Try pipeline-specific imports
from diffusers.pipelines.pipeline_utils import DiffusionPipeline, StableDiffusionMixin
print("Successfully imported pipeline utils")
from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
print("Successfully imported pipeline output")
except ImportError as e:
print("Import error:", e)
import traceback
traceback.print_exc()