Skip to content

fix bugs in trt.py line 353: current_unet is none not have cnets#331

Open
VitoChenLY wants to merge 1 commit intoNVIDIA:controlnet_v2from
VitoChenLY:controlnet_v2
Open

fix bugs in trt.py line 353: current_unet is none not have cnets#331
VitoChenLY wants to merge 1 commit intoNVIDIA:controlnet_v2from
VitoChenLY:controlnet_v2

Conversation

@VitoChenLY
Copy link

fix bugs in trt.py line 353:
ERROR:
Stable-Diffusion-WebUI-TensorRT/scripts/trt.py", line 353, in process_batch
sd_unet.current_unet.cnets = None
AttributeError: 'NoneType' object has no attribute 'cnets'

Copy link

@ice-fly ice-fly left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This is the correct logic and does fix the case where sd_unet.current_unet is not None but p has no controlnet

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants