diff --git a/stable_diffusion_videos/image_generation.py b/stable_diffusion_videos/image_generation.py index a340ccd..7a455b2 100644 --- a/stable_diffusion_videos/image_generation.py +++ b/stable_diffusion_videos/image_generation.py @@ -3,6 +3,7 @@ import json from pathlib import Path from huggingface_hub import CommitOperationAdd, create_repo, create_commit +from diffusers import __version__ as diffusers_version import torch from .upsampling import RealESRGANModel @@ -170,7 +171,7 @@ def generate_images( width=width, scheduler=dict(pipeline.scheduler.config), tiled=pipeline.tiled, - diffusers_version=pipeline.config._diffusers_version, + diffusers_version=diffusers_version, device_name=torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'unknown', ) prompt_config_path.write_text(json.dumps(cfg, indent=2, sort_keys=False))