gbddlZddlmZmZddlmZddlmZdZdefd Z Gd d eZ dS) N)ArgumentParser Namespace)logging)BaseTransformersCLICommandg-C6 ?argsc ~t|j|j|j|j|j|j|j|jS)z~ Factory function used to convert a model PyTorch checkpoint in a TensorFlow 2 checkpoint. Returns: ServeCommand ) PTtoTFCommand model_name local_dir max_error new_weightsno_prpushextra_commit_descriptionoverride_model_class)r s Z/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/commands/pt_to_tf.pyconvert_command_factoryrsA        % !   cXeZdZedefdZdedededededed ed efd Z d Z d S)r parsercB|dd}|dtdd|dtd d |d ttd td |ddd|ddd|ddd|dtd d |dtdd |t dS)z Register this command to argparse so it's available for the transformer-cli Args: parser: Root parser to register command-specific arguments zpt-to-tfzCLI tool to run convert a transformers model from a PyTorch checkpoint to a TensorFlow checkpoint. Can also be used to validate existing weights without opening PRs, with --no-pr.)helpz --model-nameTzAThe model name, including owner/organization, as seen on the hub.)typerequiredrz --local-dirzOOptional local directory of the model repository. Defaults to /tmp/{model_name})rdefaultrz --max-errorz%Maximum error tolerance. Defaults to z4. This flag should be avoided, use at your own risk.z --new-weights store_truezKOptional flag to create new TensorFlow weights, even if they already exist.)actionrz--no-prz6Optional flag to NOT open a PR with converted weights.z--pushzKOptional flag to push the weights directly to `main` (requires permissions)z--extra-commit-descriptionzXOptional additional commit description to use when opening a PR (e.g. to tag the owner).z--override-model-classNzIf you think you know better than the auto-detector, you can specify the model class here. Can be either an AutoModel class or a specific model class like BertForSequenceClassification.)func) add_parser add_argumentstrfloat MAX_ERROR set_defaultsr)r train_parsers rregister_subcommandz!PTtoTFCommand.register_subcommand,s(( d)   !! T "   !! b "   !! w www "    !! ^ "   !! l1i "    !! ^ "   !! (k "   !! $m "    !!'>!?????rr r rrrrrrc tjd|_||_|r|ntjd||_||_||_ ||_ ||_ ||_ ||_ dS)Nztransformers-cli/pt_to_tfz/tmp)r get_logger_logger _model_nameospathjoin _local_dir _max_error _new_weights_no_pr_push_extra_commit_description_override_model_class) selfr r rrrrrrr s r__init__zPTtoTFCommand.__init__ksr)*EFF %'0V))bgll6:6V6V#'  )A&%9"""rc td)Na] Converting PyTorch weights to TensorFlow weights was removed in v4.43. Instead, we recommend that you convert PyTorch weights to Safetensors, an improved format that can be loaded by any framework, including TensorFlow. For more information, please see the Safetensors conversion guide: https://huggingface.co/docs/safetensors/en/convert-weights )NotImplementedError)r8s rrunzPTtoTFCommand.runs! M   rN) __name__ __module__ __qualname__ staticmethodrr)r$r%boolr9r<rrr r +s<@N<@<@<@\<@|::: :  :  ::#&:"::::,     rr ) r.argparserrutilsrrrr&rr rBrrrEs ........((((((  )$^ ^ ^ ^ ^ .^ ^ ^ ^ ^ r