diff --git a/stable_diffusion_videos/__init__.py b/stable_diffusion_videos/__init__.py index 8943617..37ff002 100644 --- a/stable_diffusion_videos/__init__.py +++ b/stable_diffusion_videos/__init__.py @@ -114,4 +114,4 @@ def __dir__(): }, ) -__version__ = "0.7.0" +__version__ = "0.7.1" 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))