a dt@sXddlmZmZmZmZmZmZmZmZm Z ddl m Z m Z ddl mZddlmZdS)) gradgrad_and_valuevjpjvpjacrevjacfwdhessian functionalize linearize)functional_callstack_module_state)replace_all_batch_norm_modules_)vmapN)Z!torch._functorch.eager_transformsrrrrrrrr r Z torch._functorch.functional_callr r Z'torch._functorch.batch_norm_replacementr Ztorch._functorch.vmaprrr\/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/func/__init__.pys,