a d @sddlZddlZddlmZmZddlmZddlmZddl m Z ddl Z ddl m Z ddl mZmZeeZeddd d Zdd d d ZedkredS)N)ArgumentParser NamespacePath)copyfile)List)tqdm)migrate_checkpointpl_legacy_patch)argsreturnc Cs\t|j}|jdr |jn d|j}g}|sXtd|d|dtd| rf|g}| rddt j t |dd|d d D}|std |d |d tdt d|D]"}|d}|rqt||qt dt|D]L}tt|}Wdn1s,0Yt|t||qt ddS)N.z The path zr does not exist. Please provide a valid path to a checkpoint file or a directory containing checkpoints ending in cSsg|] }t|qSr).0prrw/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pytorch_lightning/utilities/upgrade_checkpoint.py ,z_upgrade..z***T) recursivez#No checkpoint files with extension z were found in z]. HINT: Try setting the `--extension` option to specify the right file extension to look for.z]Creating a backup of the existing checkpoint files before overwriting in the upgrade process.z.bakzUpgrading checkpoints ...zDone.)rpathabsolute extension startswithexists_logerrorexitis_fileis_dirglobstrinfo with_suffixrrr torchloadr save)r rrfilesfileZ backup_file checkpointrrr_upgrades@*     *r+)r cCsBtdd}|jdtdd|jddtdd d |}t|dS) NzA utility to upgrade old checkpoints to the format of the current Lightning version. This will also save a backup of the original files.) descriptionrzDPath to a checkpoint file or a directory with checkpoints to upgrade)typehelpz --extensionz-ez.ckptzRThe file extension to look for when searching for checkpoint files in a directory.)r-defaultr.)r add_argumentr" parse_argsr+)parserr rrrmainFsr3__main__)r!loggingargparserrpathlibrshutilrtypingrr%rZ%pytorch_lightning.utilities.migrationr r getLogger__name__rr+r3rrrrs     )