o !i%@s&ddlmZdejdejfddZdS)NgmreturncsLdd|jjfddd |Wd|S1swY|S)a Sets a breakpoint in `gm`'s generated python code. It drops into pdb when `gm` gets run. Args: gm: graph module to insert breakpoint. It is then recompiled for it to take effect. Returns: the `gm` with breakpoint inserted. cSs dg|S)Nzimport pdb; pdb.set_trace() bodyrrj/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torch/fx/experimental/debug.py insert_pdbs zset_trace..insert_pdbcsfddS)Ncsr|S|S)Nrr) cur_transformrrrs  z-set_trace....rr rr rr szset_trace..)Zmake_transformerN)graphZon_generate_codeZ recompile)rrr r set_traces     r)Ztorch.fxZfxZ GraphModulerrrrrs