1. 08 Aug, 2022 1 commit
  2. 22 Jul, 2022 1 commit
    • amyeroberts's avatar
      Update serving code to enable `saved_model=True` (#18153) · 8e838466
      amyeroberts authored
      
      
      * Add serving_output and serving methods to some vision models
      
      * Add serving outputs for DeiT
      
      * Don't convert hidden states - differing shapes
      
      * Make saveable
      
      * Fix up
      
      * Make swin saveable
      
      * Add in tests
      
      * Fix funnel tests (can't convert to tensor)
      
      * Fix numpy call
      
      * Tidy up a bit
      
      * Add in hidden states - resnet
      
      * Remove numpy
      
      * Fix failing tests - tensor shape and skipping tests
      
      * Remove duplicated function
      
      * PR comments - formatting and var names
      
      * PR comments
      Add suggestions made by Joao Gante:
      * Use tf.shape instead of shape_list
      * Use @tooslow decorator on tests
      * Simplify some of the logic
      
      * PR comments
      Address Yih-Dar Sheih comments - making tensor names consistent and make types float
      
      * Types consistent with docs; disable test on swin (slow)
      
      * CI trigger
      
      * Change input_features to float32
      
      * Add serving_output for segformer
      
      * Fixup
      Co-authored-by: default avatarAmy Roberts <amyeroberts@users.noreply.github.com>
      8e838466
  3. 19 Jul, 2022 1 commit
  4. 13 Jul, 2022 1 commit
  5. 12 Jul, 2022 1 commit
  6. 06 Jul, 2022 1 commit
  7. 01 Jul, 2022 1 commit
    • Matt's avatar
      XLA train step fixes (#17973) · d6cec458
      Matt authored
      * Copy inputs to train and test step before modifying them, as this breaks things
      
      * Add XLA tests, fix our loss functions to be XLA-compatible
      
      * make fixup
      
      * Update loss computation test to expect vector of per-sample losses
      
      * Patch loss for TFLED
      
      * Patch loss for TFAlbert
      
      * Add a tf_legacy_loss config flag that enables old loss functions
      
      * Stop using config.get() because it's not a dict
      
      * Skip loss computation test for RAG because its loss is very strange and I'm afraid to rewrite it
      
      * make fixup
      
      * Add XLA-compatible RAG loss
      
      * Fix dtype of loss mask for TFAlbert
      
      * Fix test for XLNet too because it overrides the default one
      
      * make fixup
      
      * Fix config test
      
      * No more depending on GPU NaN behaviour
      
      * Add test, avoid potential zero division
      
      * Fix test item assignment
      
      * Fix loss computation masking test
      
      * make fixup
      
      * Fix dtype bugs
      d6cec458
  8. 10 Jun, 2022 2 commits
  9. 01 Jun, 2022 1 commit
  10. 17 May, 2022 1 commit
  11. 12 May, 2022 1 commit
  12. 10 May, 2022 1 commit
    • Nicolas Brousse's avatar
      Add MLFLOW_FLATTEN_PARAMS support in MLflowCallback (#17148) · e99f0efe
      Nicolas Brousse authored
      * add support for MLFLOW_FLATTEN_PARAMS
      
      * ensure key is str
      
      * fix style and update warning msg
      
      * Empty commit to trigger CI
      
      * fix bug in check_inits.py
      
      * add unittest for flatten_dict utils
      
      * fix 'NoneType' object is not callable on __del__
      
      * add generic flatten_dict unittest to SPECIAL_MODULE_TO_TEST_MAP
      
      * fix style
      e99f0efe
  13. 03 May, 2022 1 commit
    • Yih-Dar's avatar
      Move test model folders (#17034) · 19420fd9
      Yih-Dar authored
      
      
      * move test model folders (TODO: fix imports and others)
      
      * fix (potentially partially) imports (in model test modules)
      
      * fix (potentially partially) imports (in tokenization test modules)
      
      * fix (potentially partially) imports (in feature extraction test modules)
      
      * fix import utils.test_modeling_tf_core
      
      * fix path ../fixtures/
      
      * fix imports about generation.test_generation_flax_utils
      
      * fix more imports
      
      * fix fixture path
      
      * fix get_test_dir
      
      * update module_to_test_file
      
      * fix get_tests_dir from wrong transformers.utils
      
      * update config.yml (CircleCI)
      
      * fix style
      
      * remove missing imports
      
      * update new model script
      
      * update check_repo
      
      * update SPECIAL_MODULE_TO_TEST_MAP
      
      * fix style
      
      * add __init__
      
      * update self-scheduled
      
      * fix add_new_model scripts
      
      * check one way to get location back
      
      * python setup.py build install
      
      * fix import in test auto
      
      * update self-scheduled.yml
      
      * update slack notification script
      
      * Add comments about artifact names
      
      * fix for yolos
      Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
      19420fd9
  14. 19 Apr, 2022 1 commit
  15. 11 Apr, 2022 1 commit
    • SaulLu's avatar
      add a warning in `SpmConverter` for sentencepiece's model using the byte fallback feature (#16629) · 1025a9b7
      SaulLu authored
      * update proto sentencepiece model
      
      * Revert "update proto sentencepiece model"
      
      This reverts commit b07f671747fec35773d0b3d4788b8b15aefa0229.
      
      * add check
      
      * add test
      
      * Revert "Revert "update proto sentencepiece model""
      
      This reverts commit 46108257b8927b73627ec8f4f3eed53a95fc700d.
      
      * test for log level
      
      * test for log level 2
      
      * warning at the warning level
      
      * clean
      
      * format
      
      * add explanation in docstring
      1025a9b7
  16. 04 Apr, 2022 1 commit
  17. 23 Mar, 2022 3 commits
  18. 17 Mar, 2022 1 commit
  19. 23 Feb, 2022 1 commit