-
Notifications
You must be signed in to change notification settings - Fork 6.7k
Closed
Description
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
Labels
No labels