Merge pull request #1702 from easydiffusion/beta

Beta
This commit is contained in:
cmdr2 2023-12-12 18:10:46 +05:30 committed by GitHub
commit dfb26ed781
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 2 deletions

View File

@ -279,8 +279,10 @@ def generate_images_internal(
delattr(pipe.unet, "_allocate_trt_buffers") delattr(pipe.unet, "_allocate_trt_buffers")
if task_data.enable_vae_tiling: if task_data.enable_vae_tiling:
if hasattr(pipe, "enable_vae_tiling"):
pipe.enable_vae_tiling() pipe.enable_vae_tiling()
else: else:
if hasattr(pipe, "disable_vae_tiling"):
pipe.disable_vae_tiling() pipe.disable_vae_tiling()
images = generate_images(context, callback=callback, **req.dict()) images = generate_images(context, callback=callback, **req.dict())

View File

@ -56,6 +56,7 @@ const SETTINGS_IDS_LIST = [
"extract_lora_from_prompt", "extract_lora_from_prompt",
"embedding-card-size-selector", "embedding-card-size-selector",
"lora_model", "lora_model",
"enable_vae_tiling",
] ]
const IGNORE_BY_DEFAULT = ["prompt"] const IGNORE_BY_DEFAULT = ["prompt"]