a d9@s6ddlZddlZddZGdddejZddZdS)NcCs0|}d|vrdddddSdddddSdS) NZvitgMb@?g?g\(\?gư>)lrZbeta1Zbeta2Zepsg+?g:0yE>)lower)Z model_namerX/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/training/params.pyget_default_paramssrc@seZdZdddZdS) ParseKwargsNc Cs`i}|D]D}|d\}}zt|||<WqtyJt|||<Yq0qt||j|dS)N=)splitast literal_eval ValueErrorstrsetattrdest)selfparser namespacevalues option_stringkwvaluekeyrrr__call__s zParseKwargs.__call__)N)__name__ __module__ __qualname__rrrrrrsrcCst}|jdtddd|jdtddd|jdtddd|jdtdd d|jd tdd d|jd gd ddd|jddddd|jdtddd|jdtddd|jdtddd|jdtdd d|jd!tdd"d|jd#td$d%d|jd&ddd'd(|jd)tdd*d|jd+td,d-d|jd.td/d0d|jd1td2d3d|jd4tdd5d|jd6tdd7d|jd8tdd9d|jd:tdd;d|jdtd?d@d|jdAtdBdCd|jdDdddEd|jdFdddGd(|jdHtdIdJd|jdKtdLdMd|jdNtdOdPd|jdQtd,dRd|jdSdddTd(|jdUtdVdWd|jdXtd,dYd|jdZdtd[d\|jd]gd^d_d`d|jdatdbdcd|jdddetdfd\|jdgdddhd|jdidddjd|jdktdldmd|jdndddod|jdptdqddrdsdt|jdutdqddvdwdt|jdxdyitdz|jd{ddd|d|jd}ddd~d|jddddd|jdtdqddd|jddddd|jddtdd\|jddddd|jddddd|jddddd|jddddd|jdtd,dd|jddtdd\|jddtdd\|jddetdd\|jddetdd\|jdtddd|jddddd|jddddd|jddddd|jddddd|jddddd|jdtdldd|jdtddd|jddddd|jdtdldmd|jddddod|jdtddd|jdtddd|jdtdOdd|jdtddd|jdtddd|jdddgddd|jddddd|jddddƍ|jddddƍ|jddddƍ||}t|j }| D]&\}}t ||durt |||q|S)Nz --train-datazyPath to file(s) with training data. When using webdataset, multiple datasources can be combined using the `::` separator.)typedefaulthelpz--train-data-upsampling-factorsaUWhen using multiple data sources with webdataset and sampling with replacement, this can be used to upsample specific data sources. Similar to --train-data, this should be a string with as many numbers as there are data sources, separated by `::` (e.g. 1::2::0.5) By default, datapoints are sampled uniformly regardless of the dataset sizes.z --val-dataz$Path to file(s) with validation dataz--train-num-sampleszTNumber of samples in dataset. Required for webdataset if not available in info file.z--val-num-sampleszRNumber of samples in dataset. Useful for webdataset if not available in info file.z--dataset-type)Z webdatasetcsvZ syntheticautor z!Which type of dataset to process.)choicesrrz--dataset-resampledF store_truezHWhether to use sampling with replacement for webdataset shard selection.)ractionrz--csv-separator z.For csv-like datasets, which separator to use.z --csv-img-keyfilepathz?For csv-like datasets, the name of the key for the image paths.z--csv-caption-keytitlez= 1.11.z--no-set-device-rankz^Don't set device index from local rank (when CUDA_VISIBLE_DEVICES restricted to one per proc).z--seedzDefault random seed.z--grad-clip-normzGradient clip.z --lock-textz,Lock full text tower by disabling gradients.z--lock-text-unlocked-layersz--lock-text-freeze-layer-normz--log-every-n-stepsdz/Log every n steps to tensorboard/console/wandb.z--coca-caption-loss-weightg@z(Weight assigned to caption loss in CoCa.z--coca-contrastive-loss-weightz7Weight assigned to contrastive loss when training CoCa.z --remote-syncz8Optinoally sync with a remote path specified by this argz--remote-sync-frequencyi,zIHow frequently to sync to a remote directly if --remote-sync is not None.z--remote-sync-protocolZs3Zfsspecz>How to do the remote sync backup if --remote-sync is not None.z--delete-previous-checkpointz     r7)r2r rActionrr7rrrrs