From 34b1ed5d550aabebdd7baad40e32d476f2a4669c Mon Sep 17 00:00:00 2001 From: GJ_HOME Date: Sat, 8 Apr 2023 00:05:52 +0800 Subject: [PATCH] fix #9234 #8970 check attr "elem_classes" before use --- modules/scripts.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules/scripts.py b/modules/scripts.py index 4d0bbd665..1c964ac2c 100644 --- a/modules/scripts.py +++ b/modules/scripts.py @@ -525,8 +525,10 @@ def add_classes_to_gradio_component(comp): """ this adds gradio-* to the component for css styling (ie gradio-button to gr.Button), as well as some others """ - - comp.elem_classes = ["gradio-" + comp.get_block_name(), *(comp.elem_classes or [])] + elem_classes = [] + if hasattr(comp, "elem_classes"): + elem_classes = comp.elem_classes + comp.elem_classes = ["gradio-" + comp.get_block_name(), *(elem_classes)] if getattr(comp, 'multiselect', False): comp.elem_classes.append('multiselect')