a dA@sddlZddlmZmZeejdgejdejdgejdgejddgejdejddgejddZejdd gejdZGd d d eZdS) N)_ButcherTableauRKAdaptiveStepsizeODESolverg?)Zdtypeg?g)alphabetaZc_solZc_errorgc@seZdZdZeZeZdS)AdaptiveHeunSolverN)__name__ __module__ __qualname__order_ADAPTIVE_HEUN_TABLEAUZtableau _AH_C_MIDZmidrrh/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torchdiffeq/_impl/adaptive_heun.pyrsr) ZtorchZ rk_commonrrZtensorfloat64r rrrrrrs&