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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions