a d<@sddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZmZdd lmZmZd d lmZmZd d lmZd d lmZGdddeZGdddeZGdddeZGdddeZdS)N)DictList) variables)create_instruction) skip_code) unimplemented) AttrSourceGlobalWeakRefSource)global_key_nameistensor) MutableLocalVariableTrackerConstantVariable)TensorVariablecseZdZdfdd ZddZddZdd Zd d Zed d dZ ddddfdd Z ddZ ddZ e ed ddZe ddZe ddZZS) ConstDictVariableNc s@tjfd|i||jt|d||_||_dS)Nrecursively_containsguards) super__init__rupdater propagatevaluesitemsuser_cls)selfrrrkwargs __class__f/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/_dynamo/variables/dicts.pyrszConstDictVariable.__init__cCsdd|jDS)NcSsi|]\}}||qSr!)as_proxy.0kvr!r!r" z.ConstDictVariable.as_proxy..rrr!r!r"r#szConstDictVariable.as_proxycCsdd|jDS)NcSsi|]\}}||qSr!)as_python_constantr$r!r!r"r(r)z8ConstDictVariable.as_python_constant..r*r+r!r!r"r,sz$ConstDictVariable.as_python_constantcCs|jSN)rr+r!r!r" python_type szConstDictVariable.python_typecCsp|jD]R\}}t|r>||jt|ddtddgn|||||j|q tdt|jgS)NT)add CALL_FUNCTIONr BUILD_MAP) rr extend_outputZcreate_load_globalr rZ append_outputcreate_load_constlen)rcodegenkeyvaluer!r!r" reconstruct#szConstDictVariable.reconstruct)argcCs|jt|||Sr-)rrget_key add_options)rr9r!r!r" getitem_const2szConstDictVariable.getitem_constList[VariableTracker]Dict[str, VariableTracker]rargsrreturnc sddlm}mt||||j}|dkr>||dS|dkrx|sN|rRJfdd|DfiS|dkr|s|rJfd d|DfiS|d kr|s|rJt |fiS|d kr |s|rJ|t |jfiS|d kr|rt |dr|j r|sHt |d ksLJt |d}t|rtt||t|}|d||<|j|dj} |dj dur| |dj ||j|| fiS|dvr*|r*t |dr*t |d|jvr*t |d kr*|dS|dkr|rt |dr|j rt|}|t |d} ||j|dfi| S|dkr |r t|dt r |j r t|}||dj|j|dj} |j|fd| i} || S|dvr`|r`t |dr`t |d|jvr`|jt |d} | S|dkr|rt |dr|t |d|jvfiSt|||SdS)Nr )r TupleVariable __getitem__rrcs4g|],\}}tj|fi|gfiqSr!r _key_to_varr$rBoptionstxr!r" Gs  z1ConstDictVariable.call_method..keyscs g|]}tj|fiqSr!rDr%r&rGrHr!r"rIZsr__len__ __setitem__r)popgetrOrr)rP __getattr__ __contains__)rrBrrrrr<rJlistr4r is_valid_key mutable_localr:r store_dict_keyr collections OrderedDictrunionr/ replace_allmodifedr;rO isinstancerr call_method) rrHnamer@rrvalr&Znewvalnew_rec_containsresultrrFr"r^5s                       zConstDictVariable.call_methodcKs|jf||d|S)z#a copy of self with different items)rr)clone)rrrrGr!r!r"r\s zConstDictVariable.modifedcs.t|g|j}fdd|D}|S)Ncs g|]}tj|fiqSr!rDrKrLr!r"rIr)z9ConstDictVariable.unpack_var_sequence..)rrrrJ)rrHr`rbr!rLr"unpack_var_sequences z%ConstDictVariable.unpack_var_sequencecCs&t|tr|jdur|jS|SdSr-)r]rspecialized_valuer,)clsr9r!r!r"r:szConstDictVariable.get_keycCs|pt|to|jduSr-)Zis_python_constantr]rre)rfr6r!r!r"rUs zConstDictVariable.is_valid_keycKsLddlm}t|r*||tt||St|s8Jt|fi|SdS)Nr )VariableBuilder)builderrgr r r r is_literal)rfrHr6rGrgr!r!r"rEs  zConstDictVariable._key_to_var)N)__name__ __module__ __qualname__rr#r,r.r8rr<r^r\rd classmethodr:rUrE __classcell__r!r!rr"rs& rcs4eZdZd fdd Zddddfdd ZZS) DefaultDictVariableNc s.tj||fi||tjus$J||_dSr-)rrrX defaultdictdefault_factory)rrrrqrrr!r"rszDefaultDictVariable.__init__r=r>rr?c sRddlm}m}t|||}|dkr.z unexpected r rz&DataClassVariable iterator constructor)rwrrrr} dataclassesfieldsinspect signaturebindapply_defaultsset argumentsrJrXrYr]r include_nonerrrir4rr) rfrr@rrGrJboundrr6r`r!r!r"create5s$     zDataClassVariable.createc Cst|}ddt|D}g}t}|D]V}t||r,t||}|j|jt |j |d|} |dusn|j rx| ||<q,| | q,|||fit ||S)NcSsg|] }|jqSr!rrr!r!r"rISr)z*DataClassVariable.wrap..)rHsource)rrrrXrYhasattrgetattrr rHr rrappendrrr) rfrhr~rrJZexcludedrr6r`varr!r!r"wrapPs(    zDataClassVariable.wrapc s(tj||fi|||s$JdSr-)rrr)rrrrGrr!r"rfszDataClassVariable.__init__cCs tdSr-)NotImplementedErrorr+r!r!r"r#jszDataClassVariable.as_proxycCsR|||jgt|j}|D]}||j|q&||tdt|gS)NCALL_FUNCTION_KW) r2Z_create_load_constrrtrrJr3rr4)rr5rJr6r!r!r"r8ms zDataClassVariable.reconstructr=r>rr?cst|||}|dkrx|s*t|dks.J|d}t|trT|j||S| |dgi |d|||Sn<|dkr|s|rJt j t |jfi|S|dkrd}t ||||S)NrCr rZto_tuple __setattr__rN)rrrr4r,r]strrr;r^rrBrTr)rrHr_r@rrGindexrr!r"r^ws$   zDataClassVariable.call_methodr_rAcs||jvr"||dt|giS|jsnddt|jD}||vrntj||sZJt|| |St ||dS)NrCcSsi|]}|j|jqSr!)r_defaultrr!r!r"r(r)z1DataClassVariable.var_getattr..) rr^rrrrrrrir;r var_getattr)rrHr_defaultsrr!r"rs zDataClassVariable.var_getattr)rjrkrl__doc__r staticmethod functools lru_cacherrrmrrrrr#r8r^rrrnr!r!rr"rws*       rwcsHeZdZdZeddZeddZfddZe dd d d Z Z S) HFPretrainedConfigVariablez/ Hack for HuggingFace PretrainedConfig cCs2zddlm}t||WSty,YdS0dS)Nr)PretrainedConfigF)Z transformers.configuration_utilsrrr)rfrr!r!r"rs    z*HFPretrainedConfigVariable.is_matching_clscCs|t|Sr-rrr!r!r"rsz-HFPretrainedConfigVariable.is_matching_objectc s.tjfi|||_|t|s*JdSr-)rrr~rr)rr~rrr!r"rsz#HFPretrainedConfigVariable.__init__rrcCsddlm}|t|j|S)Nr r)rSrrr~)rrHr_rr!r!r"rs z&HFPretrainedConfigVariable.var_getattr) rjrkrlrrrrmrrrrrnr!r!rr"rs   r) rXrrrtypingrrrSrZbytecode_transformationrZ eval_framerexcrrr r utilsr r baserrZconstantrZtensorrrrorwrr!r!r!r"s&      K7