Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

add directml python version check

Browse Source
This commit is contained in:
Vladimir Mandic
2024-06-05 17:24:22 -04:00
parent 900264a194
commit 45d1cddbce
6 changed files with 22 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/layerdiffuse/layerdiffuse_model.py
+4 -1
View File
@@ -9,9 +9,12 @@ from typing import Optional, Tuple, Union
from diffusers import AutoencoderKL
from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.models.modeling_utils import ModelMixin
from diffusers.models.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
from diffusers.models.autoencoders.vae import DecoderOutput
from diffusers.models.attention_processor import Attention, AttnProcessor
try:
from diffusers.models.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
except Exception:
from diffusers.models.unets.unet_2d_blocks import UNetMidBlock2D, get_down_block, get_up_block
def zero_module(module):