Skip to content

example script doesn't work with pip install #30

@richardrl

Description

@richardrl

The pip package doesn't include any of these:

from diffusers.modeling_utils import unwrap_model
from diffusers.optimization import get_scheduler
from diffusers.utils import logging

so "train_unconditional.py" fails:

python3 train_unconditional.py --dataset="cifar10" --resolution=64 --output_path="cifar10-ddpm" --batch_size=16 --num_epochs=100 --gradient_accumulation_steps=1 --lr=1e-4 --warmup_steps=500 --mixed_precision=no

Traceback (most recent call last): File "train_unconditional.py", line 12, in <module> from diffusers.modeling_utils import unwrap_model ImportError: cannot import name 'unwrap_model'

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions