From 031f603a04fbd1af86f324f67ad9074e1f30bb8f Mon Sep 17 00:00:00 2001 From: nateraw Date: Mon, 5 Dec 2022 18:12:41 +0000 Subject: [PATCH] :bug: fix bug introduced in 0.7.0 --- stable_diffusion_videos/image_generation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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))