keras-hub-nightly 0.15.0.dev20240823171555__tar.gz → 0.16.0.dev2024092017__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (366) hide show
  1. keras_hub_nightly-0.16.0.dev2024092017/PKG-INFO +202 -0
  2. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/README.md +30 -23
  3. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/__init__.py +0 -6
  4. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/api/__init__.py +2 -0
  5. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/api/bounding_box/__init__.py +36 -0
  6. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/api/layers/__init__.py +14 -0
  7. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/api/models/__init__.py +97 -48
  8. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/api/tokenizers/__init__.py +65 -0
  9. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/api/utils/__init__.py +22 -0
  10. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/api_export.py +15 -9
  11. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box/converters.py +529 -0
  12. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box/formats.py +162 -0
  13. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box/iou.py +263 -0
  14. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box/to_dense.py +95 -0
  15. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box/to_ragged.py +99 -0
  16. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box/utils.py +194 -0
  17. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box/validate_format.py +99 -0
  18. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/layers/preprocessing/audio_converter.py +121 -0
  19. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/layers/preprocessing/image_converter.py +130 -0
  20. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/preprocessing/masked_lm_mask_generator.py +2 -0
  21. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/preprocessing/multi_segment_packer.py +9 -8
  22. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/preprocessing/preprocessing_layer.py +2 -29
  23. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/preprocessing/random_deletion.py +33 -31
  24. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/preprocessing/random_swap.py +33 -31
  25. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/layers/preprocessing/resizing_image_converter.py +101 -0
  26. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/preprocessing/start_end_packer.py +3 -2
  27. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/albert/__init__.py +1 -2
  28. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/albert/albert_masked_lm_preprocessor.py +6 -86
  29. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/albert/albert_classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/albert/albert_text_classifier.py +34 -10
  30. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/albert/albert_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/albert/albert_text_classifier_preprocessor.py +14 -70
  31. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/albert/albert_tokenizer.py +17 -36
  32. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/backbone.py +12 -34
  33. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bart/__init__.py +1 -2
  34. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/bart/bart_seq_2_seq_lm_preprocessor.py +135 -0
  35. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bart/bart_tokenizer.py +12 -39
  36. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/bert/__init__.py +19 -0
  37. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bert/bert_masked_lm_preprocessor.py +6 -87
  38. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bert/bert_presets.py +1 -4
  39. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/bert/bert_classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/bert/bert_text_classifier.py +19 -12
  40. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/bert/bert_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/bert/bert_text_classifier_preprocessor.py +14 -70
  41. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bert/bert_tokenizer.py +17 -35
  42. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bloom/__init__.py +1 -2
  43. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/bloom/bloom_causal_lm_preprocessor.py +91 -0
  44. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bloom/bloom_tokenizer.py +12 -41
  45. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/causal_lm.py +10 -29
  46. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/causal_lm_preprocessor.py +195 -0
  47. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/csp_darknet/csp_darknet_backbone.py +54 -15
  48. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/__init__.py +1 -4
  49. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/deberta_v3_masked_lm_preprocessor.py +14 -77
  50. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/deberta_v3/deberta_v3_classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/deberta_v3/deberta_v3_text_classifier.py +16 -11
  51. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/deberta_v3/deberta_v3_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/deberta_v3/deberta_v3_text_classifier_preprocessor.py +23 -64
  52. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/deberta_v3_tokenizer.py +30 -25
  53. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/densenet/densenet_backbone.py +46 -22
  54. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/distil_bert/__init__.py +1 -4
  55. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/distil_bert/distil_bert_masked_lm_preprocessor.py +14 -76
  56. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/distil_bert/distil_bert_classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/distil_bert/distil_bert_text_classifier.py +17 -12
  57. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/distil_bert/distil_bert_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/distil_bert/distil_bert_text_classifier_preprocessor.py +23 -63
  58. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/distil_bert/distil_bert_tokenizer.py +19 -35
  59. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/efficientnet/efficientnet_backbone.py +569 -0
  60. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/efficientnet/fusedmbconv.py +229 -0
  61. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/efficientnet/mbconv.py +238 -0
  62. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/electra/__init__.py +1 -2
  63. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/electra/electra_tokenizer.py +17 -32
  64. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/f_net/__init__.py +1 -2
  65. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/f_net/f_net_masked_lm_preprocessor.py +12 -78
  66. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/f_net/f_net_classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/f_net/f_net_text_classifier.py +17 -10
  67. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/f_net/f_net_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/f_net/f_net_text_classifier_preprocessor.py +19 -63
  68. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/f_net/f_net_tokenizer.py +17 -35
  69. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/falcon/__init__.py +1 -2
  70. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/falcon/falcon_causal_lm_preprocessor.py +90 -0
  71. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/falcon/falcon_tokenizer.py +12 -35
  72. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/__init__.py +1 -2
  73. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/gemma/gemma_causal_lm_preprocessor.py +83 -0
  74. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/gemma_decoder_block.py +1 -1
  75. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/gemma_tokenizer.py +12 -23
  76. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt2/__init__.py +1 -2
  77. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/gpt2/gpt2_causal_lm_preprocessor.py +90 -0
  78. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/gpt_neo_x/gpt_neo_x_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/gpt2/gpt2_preprocessor.py +17 -53
  79. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt2/gpt2_tokenizer.py +12 -34
  80. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor.py +56 -0
  81. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt_neo_x/gpt_neo_x_tokenizer.py +12 -34
  82. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/image_classifier.py +0 -5
  83. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/image_classifier_preprocessor.py +83 -0
  84. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/__init__.py +1 -2
  85. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_causal_lm_preprocessor.py +6 -85
  86. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_tokenizer.py +12 -25
  87. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama3/__init__.py +1 -2
  88. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/llama3/llama3_causal_lm_preprocessor.py +90 -0
  89. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/llama3/llama3_tokenizer.py +42 -0
  90. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/masked_lm.py +0 -2
  91. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/masked_lm_preprocessor.py +156 -0
  92. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/__init__.py +1 -2
  93. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/mistral/mistral_causal_lm_preprocessor.py +90 -0
  94. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/mistral_tokenizer.py +12 -23
  95. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mix_transformer/mix_transformer_backbone.py +2 -2
  96. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/mobilenet/mobilenet_backbone.py +530 -0
  97. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/mobilenet/mobilenet_image_classifier.py +114 -0
  98. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/opt/__init__.py +1 -2
  99. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/opt/opt_causal_lm_preprocessor.py +6 -93
  100. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/opt/opt_tokenizer.py +12 -41
  101. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/__init__.py +1 -4
  102. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/pali_gemma_causal_lm_preprocessor.py +28 -28
  103. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/pali_gemma/pali_gemma_image_converter.py +25 -0
  104. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/pali_gemma_presets.py +5 -5
  105. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/pali_gemma_tokenizer.py +11 -3
  106. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/__init__.py +1 -2
  107. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_causal_lm.py +3 -9
  108. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_causal_lm_preprocessor.py +6 -89
  109. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_tokenizer.py +12 -36
  110. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/preprocessor.py +72 -83
  111. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/resnet/__init__.py +19 -0
  112. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/resnet/resnet_backbone.py +390 -42
  113. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/resnet/resnet_image_classifier.py +33 -6
  114. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/resnet/resnet_image_classifier_preprocessor.py +28 -0
  115. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/llama3/llama3_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/resnet/resnet_image_converter.py +7 -5
  116. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/resnet/resnet_presets.py +95 -0
  117. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/retinanet/anchor_generator.py +175 -0
  118. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/retinanet/box_matcher.py +259 -0
  119. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/retinanet/non_max_supression.py +578 -0
  120. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/roberta/__init__.py +1 -2
  121. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/roberta/roberta_masked_lm_preprocessor.py +22 -74
  122. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/roberta/roberta_classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/roberta/roberta_text_classifier.py +16 -11
  123. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/roberta/roberta_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/roberta/roberta_text_classifier_preprocessor.py +21 -53
  124. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/roberta/roberta_tokenizer.py +13 -52
  125. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/seq_2_seq_lm_preprocessor.py +269 -0
  126. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/clip_encoder_block.py +103 -0
  127. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/clip_preprocessor.py +93 -0
  128. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/clip_text_encoder.py +149 -0
  129. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/clip_tokenizer.py +167 -0
  130. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/mmdit.py +427 -0
  131. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/mmdit_block.py +317 -0
  132. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/t5_xxl_preprocessor.py +74 -0
  133. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/t5_xxl_text_encoder.py +155 -0
  134. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/vae_attention.py +126 -0
  135. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3/vae_image_decoder.py +186 -0
  136. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/t5/__init__.py +1 -2
  137. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/t5/t5_tokenizer.py +13 -23
  138. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/task.py +71 -116
  139. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/text_classifier.py +19 -13
  140. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/text_classifier_preprocessor.py +138 -0
  141. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/whisper/__init__.py +1 -2
  142. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/whisper/whisper_audio_feature_extractor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/whisper/whisper_audio_converter.py +20 -18
  143. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/whisper/whisper_backbone.py +0 -3
  144. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/whisper/whisper_presets.py +10 -10
  145. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/whisper/whisper_tokenizer.py +20 -16
  146. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlm_roberta/__init__.py +1 -4
  147. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor.py +26 -72
  148. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/xlm_roberta/xlm_roberta_classifier.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/xlm_roberta/xlm_roberta_text_classifier.py +16 -11
  149. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/xlm_roberta/xlm_roberta_preprocessor.py → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/xlm_roberta/xlm_roberta_text_classifier_preprocessor.py +26 -53
  150. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlm_roberta/xlm_roberta_tokenizer.py +25 -10
  151. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tests/test_case.py +46 -0
  152. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/tokenizers/__init__.py +13 -0
  153. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/byte_pair_tokenizer.py +30 -17
  154. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/byte_tokenizer.py +14 -15
  155. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/sentence_piece_tokenizer.py +20 -7
  156. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/tokenizer.py +67 -32
  157. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/unicode_codepoint_tokenizer.py +14 -15
  158. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/word_piece_tokenizer.py +34 -47
  159. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/__init__.py +13 -0
  160. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/imagenet/__init__.py +13 -0
  161. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/imagenet/imagenet_utils.py +1067 -0
  162. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/keras_utils.py +0 -50
  163. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/preset_utils.py +230 -68
  164. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/tensor_utils.py +324 -0
  165. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/timm/__init__.py +13 -0
  166. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/timm/convert_resnet.py +19 -16
  167. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/timm/preset_loader.py +66 -0
  168. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/transformers/__init__.py +13 -0
  169. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/transformers/convert_albert.py +193 -0
  170. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/transformers/convert_bart.py +373 -0
  171. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/transformers/convert_bert.py +7 -17
  172. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/transformers/convert_distilbert.py +10 -20
  173. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/transformers/convert_gemma.py +5 -19
  174. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/transformers/convert_gpt2.py +5 -18
  175. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/transformers/convert_llama3.py +7 -18
  176. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/transformers/convert_mistral.py +129 -0
  177. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/transformers/convert_pali_gemma.py +7 -29
  178. keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/utils/transformers/preset_loader.py +77 -0
  179. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/transformers/safetensor_utils.py +2 -2
  180. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/version_utils.py +1 -1
  181. keras_hub_nightly-0.16.0.dev2024092017/keras_hub_nightly.egg-info/PKG-INFO +202 -0
  182. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub_nightly.egg-info/SOURCES.txt +67 -30
  183. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/setup.py +7 -8
  184. keras_hub_nightly-0.15.0.dev20240823171555/PKG-INFO +0 -34
  185. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/api/tokenizers/__init__.py +0 -35
  186. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/bart/bart_preprocessor.py +0 -276
  187. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/bart/bart_seq_2_seq_lm_preprocessor.py +0 -262
  188. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/bert/__init__.py +0 -23
  189. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/bloom/bloom_causal_lm_preprocessor.py +0 -176
  190. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/bloom/bloom_preprocessor.py +0 -185
  191. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/electra/electra_preprocessor.py +0 -154
  192. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/falcon/falcon_causal_lm_preprocessor.py +0 -173
  193. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/falcon/falcon_preprocessor.py +0 -187
  194. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/gemma/gemma_causal_lm_preprocessor.py +0 -167
  195. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/gemma/gemma_preprocessor.py +0 -191
  196. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/gpt2/gpt2_causal_lm_preprocessor.py +0 -173
  197. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/gpt2/gpt2_preprocessor.py +0 -187
  198. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor.py +0 -141
  199. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/llama/llama_preprocessor.py +0 -189
  200. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/llama3/llama3_causal_lm_preprocessor.py +0 -173
  201. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/llama3/llama3_tokenizer.py +0 -63
  202. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/mistral/mistral_causal_lm_preprocessor.py +0 -175
  203. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/mistral/mistral_preprocessor.py +0 -190
  204. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/opt/opt_preprocessor.py +0 -188
  205. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/phi3/phi3_preprocessor.py +0 -190
  206. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/whisper/whisper_preprocessor.py +0 -326
  207. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/utils/tensor_utils.py +0 -206
  208. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/utils/timm/convert.py +0 -37
  209. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/utils/transformers/convert.py +0 -101
  210. keras_hub_nightly-0.15.0.dev20240823171555/keras_hub_nightly.egg-info/PKG-INFO +0 -34
  211. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/api/metrics/__init__.py +0 -0
  212. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/api/samplers/__init__.py +0 -0
  213. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/__init__.py +0 -0
  214. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/layers → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/bounding_box}/__init__.py +0 -0
  215. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/layers/modeling → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/layers}/__init__.py +0 -0
  216. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/layers/preprocessing → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/layers/modeling}/__init__.py +0 -0
  217. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/alibi_bias.py +0 -0
  218. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/cached_multi_head_attention.py +0 -0
  219. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/f_net_encoder.py +0 -0
  220. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/masked_lm_head.py +0 -0
  221. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/position_embedding.py +0 -0
  222. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/reversible_embedding.py +0 -0
  223. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/rotary_embedding.py +0 -0
  224. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/sine_position_encoding.py +0 -0
  225. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/token_and_position_embedding.py +0 -0
  226. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/transformer_decoder.py +0 -0
  227. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/transformer_encoder.py +0 -0
  228. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/layers/modeling/transformer_layer_utils.py +0 -0
  229. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/metrics → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/layers/preprocessing}/__init__.py +0 -0
  230. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/metrics}/__init__.py +0 -0
  231. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/metrics/bleu.py +0 -0
  232. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/metrics/edit_distance.py +0 -0
  233. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/metrics/perplexity.py +0 -0
  234. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/metrics/rouge_base.py +0 -0
  235. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/metrics/rouge_l.py +0 -0
  236. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/metrics/rouge_n.py +0 -0
  237. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/csp_darknet → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models}/__init__.py +0 -0
  238. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/albert/albert_backbone.py +0 -0
  239. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/albert/albert_masked_lm.py +0 -0
  240. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/albert/albert_presets.py +0 -0
  241. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bart/bart_backbone.py +0 -0
  242. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bart/bart_presets.py +0 -0
  243. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bart/bart_seq_2_seq_lm.py +0 -0
  244. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bert/bert_backbone.py +0 -0
  245. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bert/bert_masked_lm.py +0 -0
  246. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bloom/bloom_attention.py +0 -0
  247. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bloom/bloom_backbone.py +0 -0
  248. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bloom/bloom_causal_lm.py +0 -0
  249. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bloom/bloom_decoder.py +0 -0
  250. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/bloom/bloom_presets.py +0 -0
  251. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/densenet → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/csp_darknet}/__init__.py +0 -0
  252. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/csp_darknet/csp_darknet_image_classifier.py +0 -0
  253. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/deberta_v3_backbone.py +0 -0
  254. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/deberta_v3_masked_lm.py +0 -0
  255. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/deberta_v3_presets.py +0 -0
  256. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/disentangled_attention_encoder.py +0 -0
  257. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/disentangled_self_attention.py +0 -0
  258. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/deberta_v3/relative_embedding.py +0 -0
  259. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/gpt_neo_x → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/densenet}/__init__.py +0 -0
  260. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/densenet/densenet_image_classifier.py +0 -0
  261. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/distil_bert/distil_bert_backbone.py +0 -0
  262. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/distil_bert/distil_bert_masked_lm.py +0 -0
  263. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/distil_bert/distil_bert_presets.py +0 -0
  264. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/mix_transformer → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/efficientnet}/__init__.py +0 -0
  265. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/electra/electra_backbone.py +0 -0
  266. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/electra/electra_presets.py +0 -0
  267. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/f_net/f_net_backbone.py +0 -0
  268. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/f_net/f_net_masked_lm.py +0 -0
  269. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/f_net/f_net_presets.py +0 -0
  270. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/falcon/falcon_attention.py +0 -0
  271. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/falcon/falcon_backbone.py +0 -0
  272. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/falcon/falcon_causal_lm.py +0 -0
  273. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/falcon/falcon_presets.py +0 -0
  274. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/falcon/falcon_transformer_decoder.py +0 -0
  275. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/feature_pyramid_backbone.py +0 -0
  276. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/gemma_attention.py +0 -0
  277. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/gemma_backbone.py +0 -0
  278. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/gemma_causal_lm.py +0 -0
  279. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/gemma_presets.py +0 -0
  280. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gemma/rms_normalization.py +0 -0
  281. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt2/gpt2_backbone.py +0 -0
  282. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt2/gpt2_causal_lm.py +0 -0
  283. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt2/gpt2_presets.py +0 -0
  284. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/resnet → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/gpt_neo_x}/__init__.py +0 -0
  285. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt_neo_x/gpt_neo_x_attention.py +0 -0
  286. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt_neo_x/gpt_neo_x_backbone.py +0 -0
  287. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt_neo_x/gpt_neo_x_causal_lm.py +0 -0
  288. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/gpt_neo_x/gpt_neo_x_decoder.py +0 -0
  289. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_attention.py +0 -0
  290. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_backbone.py +0 -0
  291. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_causal_lm.py +0 -0
  292. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_decoder.py +0 -0
  293. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_layernorm.py +0 -0
  294. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama/llama_presets.py +0 -0
  295. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama3/llama3_backbone.py +0 -0
  296. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama3/llama3_causal_lm.py +0 -0
  297. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/llama3/llama3_presets.py +0 -0
  298. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/mistral_attention.py +0 -0
  299. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/mistral_backbone.py +0 -0
  300. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/mistral_causal_lm.py +0 -0
  301. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/mistral_layer_norm.py +0 -0
  302. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/mistral_presets.py +0 -0
  303. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mistral/mistral_transformer_decoder.py +0 -0
  304. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/vgg → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/mix_transformer}/__init__.py +0 -0
  305. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mix_transformer/mix_transformer_classifier.py +0 -0
  306. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/mix_transformer/mix_transformer_layers.py +0 -0
  307. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/vit_det → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/mobilenet}/__init__.py +0 -0
  308. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/opt/opt_backbone.py +0 -0
  309. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/opt/opt_causal_lm.py +0 -0
  310. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/opt/opt_presets.py +0 -0
  311. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/pali_gemma_backbone.py +0 -0
  312. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/pali_gemma_causal_lm.py +0 -0
  313. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/pali_gemma_decoder_block.py +0 -0
  314. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/pali_gemma/pali_gemma_vit.py +0 -0
  315. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_attention.py +0 -0
  316. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_backbone.py +0 -0
  317. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_decoder.py +0 -0
  318. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_layernorm.py +0 -0
  319. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_presets.py +0 -0
  320. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/phi3/phi3_rotary_embedding.py +0 -0
  321. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/models/xlnet → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/retinanet}/__init__.py +0 -0
  322. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/roberta/roberta_backbone.py +0 -0
  323. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/roberta/roberta_masked_lm.py +0 -0
  324. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/roberta/roberta_presets.py +0 -0
  325. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/seq_2_seq_lm.py +0 -0
  326. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/samplers → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/stable_diffusion_v3}/__init__.py +0 -0
  327. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/t5/t5_backbone.py +0 -0
  328. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/t5/t5_layer_norm.py +0 -0
  329. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/t5/t5_multi_head_attention.py +0 -0
  330. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/t5/t5_presets.py +0 -0
  331. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/t5/t5_transformer_layer.py +0 -0
  332. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/tests → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/vgg}/__init__.py +0 -0
  333. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/vgg/vgg_backbone.py +0 -0
  334. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/vgg/vgg_image_classifier.py +0 -0
  335. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/tokenizers → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/vit_det}/__init__.py +0 -0
  336. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/vit_det/vit_det_backbone.py +0 -0
  337. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/vit_det/vit_layers.py +0 -0
  338. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/whisper/whisper_cached_multi_head_attention.py +0 -0
  339. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/whisper/whisper_decoder.py +0 -0
  340. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/whisper/whisper_encoder.py +0 -0
  341. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlm_roberta/xlm_roberta_backbone.py +0 -0
  342. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlm_roberta/xlm_roberta_masked_lm.py +0 -0
  343. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlm_roberta/xlm_roberta_presets.py +0 -0
  344. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/utils → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/models/xlnet}/__init__.py +0 -0
  345. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlnet/relative_attention.py +0 -0
  346. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlnet/xlnet_backbone.py +0 -0
  347. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlnet/xlnet_content_and_query_embedding.py +0 -0
  348. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/models/xlnet/xlnet_encoder.py +0 -0
  349. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/utils/timm → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/samplers}/__init__.py +0 -0
  350. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/beam_sampler.py +0 -0
  351. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/contrastive_sampler.py +0 -0
  352. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/greedy_sampler.py +0 -0
  353. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/random_sampler.py +0 -0
  354. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/sampler.py +0 -0
  355. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/serialization.py +0 -0
  356. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/top_k_sampler.py +0 -0
  357. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/samplers/top_p_sampler.py +0 -0
  358. {keras_hub_nightly-0.15.0.dev20240823171555/keras_hub/src/utils/transformers → keras_hub_nightly-0.16.0.dev2024092017/keras_hub/src/tests}/__init__.py +0 -0
  359. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/sentence_piece_tokenizer_trainer.py +0 -0
  360. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/tokenizers/word_piece_tokenizer_trainer.py +0 -0
  361. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/pipeline_model.py +0 -0
  362. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub/src/utils/python_utils.py +0 -0
  363. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub_nightly.egg-info/dependency_links.txt +0 -0
  364. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub_nightly.egg-info/requires.txt +0 -0
  365. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/keras_hub_nightly.egg-info/top_level.txt +0 -0
  366. {keras_hub_nightly-0.15.0.dev20240823171555 → keras_hub_nightly-0.16.0.dev2024092017}/setup.cfg +0 -0
@@ -0,0 +1,202 @@
1
+ Metadata-Version: 2.1
2
+ Name: keras_hub-nightly
3
+ Version: 0.16.0.dev2024092017
4
+ Summary: Industry-strength Natural Language Processing extensions for Keras.
5
+ Home-page: https://github.com/keras-team/keras-hub
6
+ Author: Keras team
7
+ Author-email: keras-hub@google.com
8
+ License: Apache License 2.0
9
+ Classifier: Development Status :: 3 - Alpha
10
+ Classifier: Programming Language :: Python :: 3
11
+ Classifier: Programming Language :: Python :: 3.9
12
+ Classifier: Programming Language :: Python :: 3.10
13
+ Classifier: Programming Language :: Python :: 3.11
14
+ Classifier: Programming Language :: Python :: 3 :: Only
15
+ Classifier: Operating System :: Unix
16
+ Classifier: Operating System :: Microsoft :: Windows
17
+ Classifier: Operating System :: MacOS
18
+ Classifier: Intended Audience :: Science/Research
19
+ Classifier: Topic :: Scientific/Engineering
20
+ Classifier: Topic :: Software Development
21
+ Requires-Python: >=3.9
22
+ Description-Content-Type: text/markdown
23
+ Requires-Dist: absl-py
24
+ Requires-Dist: numpy
25
+ Requires-Dist: packaging
26
+ Requires-Dist: regex
27
+ Requires-Dist: rich
28
+ Requires-Dist: kagglehub
29
+ Requires-Dist: tensorflow-text; platform_system != "Darwin"
30
+ Provides-Extra: extras
31
+ Requires-Dist: rouge-score; extra == "extras"
32
+ Requires-Dist: sentencepiece; extra == "extras"
33
+
34
+ # KerasNLP: Multi-framework NLP Models
35
+ [![](https://github.com/keras-team/keras-hub/workflows/Tests/badge.svg?branch=master)](https://github.com/keras-team/keras-hub/actions?query=workflow%3ATests+branch%3Amaster)
36
+ ![Python](https://img.shields.io/badge/python-v3.9.0+-success.svg)
37
+ [![contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat)](https://github.com/keras-team/keras-hub/issues)
38
+
39
+ > [!IMPORTANT]
40
+ > 📢 KerasNLP is becoming KerasHub! 📢 Read
41
+ > [the announcement](https://github.com/keras-team/keras-hub/issues/1831).
42
+ >
43
+ > We have renamed the repo to KerasHub in preparation for the release, but have not yet
44
+ > released the new package. Follow the announcement for news.
45
+
46
+ KerasNLP is a natural language processing library that works natively
47
+ with TensorFlow, JAX, or PyTorch. KerasNLP provides a repository of pre-trained
48
+ models and a collection of lower-level building blocks for language modeling.
49
+ Built on Keras 3, models can be trained and serialized in any framework
50
+ and re-used in another without costly migrations.
51
+
52
+ This library is an extension of the core Keras API; all high-level modules are
53
+ Layers and Models that receive that same level of polish as core Keras.
54
+ If you are familiar with Keras, congratulations! You already understand most of
55
+ KerasNLP.
56
+
57
+ All models support JAX, TensorFlow, and PyTorch from a single model
58
+ definition and can be fine-tuned on GPUs and TPUs out of the box. Models can
59
+ be trained on individual accelerators with built-in PEFT techniques, or
60
+ fine-tuned at scale with model and data parallel training. See our
61
+ [Getting Started guide](https://keras.io/guides/keras_nlp/getting_started)
62
+ to start learning our API. Browse our models on
63
+ [Kaggle](https://www.kaggle.com/organizations/keras/models).
64
+ We welcome contributions.
65
+
66
+ ## Quick Links
67
+
68
+ ### For everyone
69
+
70
+ - [Home Page](https://keras.io/keras_nlp)
71
+ - [Developer Guides](https://keras.io/guides/keras_nlp)
72
+ - [API Reference](https://keras.io/api/keras_nlp)
73
+ - [Pre-trained Models](https://www.kaggle.com/organizations/keras/models)
74
+
75
+ ### For contributors
76
+
77
+ - [Contributing Guide](CONTRIBUTING.md)
78
+ - [Roadmap](ROADMAP.md)
79
+ - [Style Guide](STYLE_GUIDE.md)
80
+ - [API Design Guide](API_DESIGN_GUIDE.md)
81
+ - [Call for Contributions](https://github.com/keras-team/keras-hub/issues?q=is%3Aissue+is%3Aopen+label%3A%22contributions+welcome%22)
82
+
83
+ ## Quickstart
84
+
85
+ Fine-tune BERT on IMDb movie reviews:
86
+
87
+ ```python
88
+ import os
89
+ os.environ["KERAS_BACKEND"] = "jax" # Or "tensorflow" or "torch"!
90
+
91
+ import keras_nlp
92
+ import tensorflow_datasets as tfds
93
+
94
+ imdb_train, imdb_test = tfds.load(
95
+ "imdb_reviews",
96
+ split=["train", "test"],
97
+ as_supervised=True,
98
+ batch_size=16,
99
+ )
100
+ # Load a BERT model.
101
+ classifier = keras_nlp.models.Classifier.from_preset(
102
+ "bert_base_en",
103
+ num_classes=2,
104
+ activation="softmax",
105
+ )
106
+ # Fine-tune on IMDb movie reviews.
107
+ classifier.fit(imdb_train, validation_data=imdb_test)
108
+ # Predict two new examples.
109
+ classifier.predict(["What an amazing movie!", "A total waste of my time."])
110
+ ```
111
+
112
+ Try it out [in a colab](https://colab.research.google.com/gist/mattdangerw/e457e42d5ea827110c8d5cb4eb9d9a07/kerasnlp-quickstart.ipynb).
113
+ For more in depth guides and examples, visit
114
+ [keras.io/keras_nlp](https://keras.io/keras_nlp/).
115
+
116
+ ## Installation
117
+
118
+ To install the latest KerasNLP release with Keras 3, simply run:
119
+
120
+ ```
121
+ pip install --upgrade keras-nlp
122
+ ```
123
+
124
+ To install the latest nightly changes for both KerasNLP and Keras, you can use
125
+ our nightly package.
126
+
127
+ ```
128
+ pip install --upgrade keras-nlp-nightly
129
+ ```
130
+
131
+ Note that currently, installing KerasNLP will always pull in TensorFlow for use
132
+ of the `tf.data` API for preprocessing. Even when pre-processing with `tf.data`,
133
+ training can still happen on any backend.
134
+
135
+ Read [Getting started with Keras](https://keras.io/getting_started/) for more
136
+ information on installing Keras 3 and compatibility with different frameworks.
137
+
138
+ > [!IMPORTANT]
139
+ > We recommend using KerasNLP with TensorFlow 2.16 or later, as TF 2.16 packages
140
+ > Keras 3 by default.
141
+
142
+ ## Configuring your backend
143
+
144
+ If you have Keras 3 installed in your environment (see installation above),
145
+ you can use KerasNLP with any of JAX, TensorFlow and PyTorch. To do so, set the
146
+ `KERAS_BACKEND` environment variable. For example:
147
+
148
+ ```shell
149
+ export KERAS_BACKEND=jax
150
+ ```
151
+
152
+ Or in Colab, with:
153
+
154
+ ```python
155
+ import os
156
+ os.environ["KERAS_BACKEND"] = "jax"
157
+
158
+ import keras_nlp
159
+ ```
160
+
161
+ > [!IMPORTANT]
162
+ > Make sure to set the `KERAS_BACKEND` before import any Keras libraries, it
163
+ > will be used to set up Keras when it is first imported.
164
+
165
+ ## Compatibility
166
+
167
+ We follow [Semantic Versioning](https://semver.org/), and plan to
168
+ provide backwards compatibility guarantees both for code and saved models built
169
+ with our components. While we continue with pre-release `0.y.z` development, we
170
+ may break compatibility at any time and APIs should not be consider stable.
171
+
172
+ ## Disclaimer
173
+
174
+ KerasNLP provides access to pre-trained models via the `keras_nlp.models` API.
175
+ These pre-trained models are provided on an "as is" basis, without warranties
176
+ or conditions of any kind. The following underlying models are provided by third
177
+ parties, and subject to separate licenses:
178
+ BART, BLOOM, DeBERTa, DistilBERT, GPT-2, Llama, Mistral, OPT, RoBERTa, Whisper,
179
+ and XLM-RoBERTa.
180
+
181
+ ## Citing KerasNLP
182
+
183
+ If KerasNLP helps your research, we appreciate your citations.
184
+ Here is the BibTeX entry:
185
+
186
+ ```bibtex
187
+ @misc{kerasnlp2022,
188
+ title={KerasNLP},
189
+ author={Watson, Matthew, and Qian, Chen, and Bischof, Jonathan and Chollet,
190
+ Fran\c{c}ois and others},
191
+ year={2022},
192
+ howpublished={\url{https://github.com/keras-team/keras-hub}},
193
+ }
194
+ ```
195
+
196
+ ## Acknowledgements
197
+
198
+ Thank you to all of our wonderful contributors!
199
+
200
+ <a href="https://github.com/keras-team/keras-hub/graphs/contributors">
201
+ <img src="https://contrib.rocks/image?repo=keras-team/keras-hub" />
202
+ </a>
@@ -1,10 +1,17 @@
1
- # KerasHub: Multi-framework NLP Models
1
+ # KerasNLP: Multi-framework NLP Models
2
2
  [![](https://github.com/keras-team/keras-hub/workflows/Tests/badge.svg?branch=master)](https://github.com/keras-team/keras-hub/actions?query=workflow%3ATests+branch%3Amaster)
3
3
  ![Python](https://img.shields.io/badge/python-v3.9.0+-success.svg)
4
4
  [![contributions welcome](https://img.shields.io/badge/contributions-welcome-brightgreen.svg?style=flat)](https://github.com/keras-team/keras-hub/issues)
5
5
 
6
- KerasHub is a natural language processing library that works natively
7
- with TensorFlow, JAX, or PyTorch. KerasHub provides a repository of pre-trained
6
+ > [!IMPORTANT]
7
+ > 📢 KerasNLP is becoming KerasHub! 📢 Read
8
+ > [the announcement](https://github.com/keras-team/keras-hub/issues/1831).
9
+ >
10
+ > We have renamed the repo to KerasHub in preparation for the release, but have not yet
11
+ > released the new package. Follow the announcement for news.
12
+
13
+ KerasNLP is a natural language processing library that works natively
14
+ with TensorFlow, JAX, or PyTorch. KerasNLP provides a repository of pre-trained
8
15
  models and a collection of lower-level building blocks for language modeling.
9
16
  Built on Keras 3, models can be trained and serialized in any framework
10
17
  and re-used in another without costly migrations.
@@ -12,13 +19,13 @@ and re-used in another without costly migrations.
12
19
  This library is an extension of the core Keras API; all high-level modules are
13
20
  Layers and Models that receive that same level of polish as core Keras.
14
21
  If you are familiar with Keras, congratulations! You already understand most of
15
- KerasHub.
22
+ KerasNLP.
16
23
 
17
24
  All models support JAX, TensorFlow, and PyTorch from a single model
18
25
  definition and can be fine-tuned on GPUs and TPUs out of the box. Models can
19
26
  be trained on individual accelerators with built-in PEFT techniques, or
20
27
  fine-tuned at scale with model and data parallel training. See our
21
- [Getting Started guide](https://keras.io/guides/keras_hub/getting_started)
28
+ [Getting Started guide](https://keras.io/guides/keras_nlp/getting_started)
22
29
  to start learning our API. Browse our models on
23
30
  [Kaggle](https://www.kaggle.com/organizations/keras/models).
24
31
  We welcome contributions.
@@ -27,9 +34,9 @@ We welcome contributions.
27
34
 
28
35
  ### For everyone
29
36
 
30
- - [Home Page](https://keras.io/keras_hub)
31
- - [Developer Guides](https://keras.io/guides/keras_hub)
32
- - [API Reference](https://keras.io/api/keras_hub)
37
+ - [Home Page](https://keras.io/keras_nlp)
38
+ - [Developer Guides](https://keras.io/guides/keras_nlp)
39
+ - [API Reference](https://keras.io/api/keras_nlp)
33
40
  - [Pre-trained Models](https://www.kaggle.com/organizations/keras/models)
34
41
 
35
42
  ### For contributors
@@ -48,7 +55,7 @@ Fine-tune BERT on IMDb movie reviews:
48
55
  import os
49
56
  os.environ["KERAS_BACKEND"] = "jax" # Or "tensorflow" or "torch"!
50
57
 
51
- import keras_hub
58
+ import keras_nlp
52
59
  import tensorflow_datasets as tfds
53
60
 
54
61
  imdb_train, imdb_test = tfds.load(
@@ -58,7 +65,7 @@ imdb_train, imdb_test = tfds.load(
58
65
  batch_size=16,
59
66
  )
60
67
  # Load a BERT model.
61
- classifier = keras_hub.models.Classifier.from_preset(
68
+ classifier = keras_nlp.models.Classifier.from_preset(
62
69
  "bert_base_en",
63
70
  num_classes=2,
64
71
  activation="softmax",
@@ -71,24 +78,24 @@ classifier.predict(["What an amazing movie!", "A total waste of my time."])
71
78
 
72
79
  Try it out [in a colab](https://colab.research.google.com/gist/mattdangerw/e457e42d5ea827110c8d5cb4eb9d9a07/kerasnlp-quickstart.ipynb).
73
80
  For more in depth guides and examples, visit
74
- [keras.io/keras_hub](https://keras.io/keras_hub/).
81
+ [keras.io/keras_nlp](https://keras.io/keras_nlp/).
75
82
 
76
83
  ## Installation
77
84
 
78
- To install the latest KerasHub release with Keras 3, simply run:
85
+ To install the latest KerasNLP release with Keras 3, simply run:
79
86
 
80
87
  ```
81
- pip install --upgrade keras-hub
88
+ pip install --upgrade keras-nlp
82
89
  ```
83
90
 
84
- To install the latest nightly changes for both KerasHub and Keras, you can use
91
+ To install the latest nightly changes for both KerasNLP and Keras, you can use
85
92
  our nightly package.
86
93
 
87
94
  ```
88
- pip install --upgrade keras-hub-nightly
95
+ pip install --upgrade keras-nlp-nightly
89
96
  ```
90
97
 
91
- Note that currently, installing KerasHub will always pull in TensorFlow for use
98
+ Note that currently, installing KerasNLP will always pull in TensorFlow for use
92
99
  of the `tf.data` API for preprocessing. Even when pre-processing with `tf.data`,
93
100
  training can still happen on any backend.
94
101
 
@@ -96,13 +103,13 @@ Read [Getting started with Keras](https://keras.io/getting_started/) for more
96
103
  information on installing Keras 3 and compatibility with different frameworks.
97
104
 
98
105
  > [!IMPORTANT]
99
- > We recommend using KerasHub with TensorFlow 2.16 or later, as TF 2.16 packages
106
+ > We recommend using KerasNLP with TensorFlow 2.16 or later, as TF 2.16 packages
100
107
  > Keras 3 by default.
101
108
 
102
109
  ## Configuring your backend
103
110
 
104
111
  If you have Keras 3 installed in your environment (see installation above),
105
- you can use KerasHub with any of JAX, TensorFlow and PyTorch. To do so, set the
112
+ you can use KerasNLP with any of JAX, TensorFlow and PyTorch. To do so, set the
106
113
  `KERAS_BACKEND` environment variable. For example:
107
114
 
108
115
  ```shell
@@ -115,7 +122,7 @@ Or in Colab, with:
115
122
  import os
116
123
  os.environ["KERAS_BACKEND"] = "jax"
117
124
 
118
- import keras_hub
125
+ import keras_nlp
119
126
  ```
120
127
 
121
128
  > [!IMPORTANT]
@@ -131,21 +138,21 @@ may break compatibility at any time and APIs should not be consider stable.
131
138
 
132
139
  ## Disclaimer
133
140
 
134
- KerasHub provides access to pre-trained models via the `keras_hub.models` API.
141
+ KerasNLP provides access to pre-trained models via the `keras_nlp.models` API.
135
142
  These pre-trained models are provided on an "as is" basis, without warranties
136
143
  or conditions of any kind. The following underlying models are provided by third
137
144
  parties, and subject to separate licenses:
138
145
  BART, BLOOM, DeBERTa, DistilBERT, GPT-2, Llama, Mistral, OPT, RoBERTa, Whisper,
139
146
  and XLM-RoBERTa.
140
147
 
141
- ## Citing KerasHub
148
+ ## Citing KerasNLP
142
149
 
143
- If KerasHub helps your research, we appreciate your citations.
150
+ If KerasNLP helps your research, we appreciate your citations.
144
151
  Here is the BibTeX entry:
145
152
 
146
153
  ```bibtex
147
154
  @misc{kerasnlp2022,
148
- title={KerasHub},
155
+ title={KerasNLP},
149
156
  author={Watson, Matthew, and Qian, Chen, and Bischof, Jonathan and Chollet,
150
157
  Fran\c{c}ois and others},
151
158
  year={2022},
@@ -11,12 +11,6 @@
11
11
  # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
12
  # See the License for the specific language governing permissions and
13
13
  # limitations under the License.
14
- """DO NOT EDIT.
15
-
16
- This file was autogenerated. Do not edit it by hand,
17
- since your modifications would be overwritten.
18
- """
19
-
20
14
  import os
21
15
 
22
16
  # sentencepiece segfaults on some version of tensorflow if tf is imported first.
@@ -17,11 +17,13 @@ This file was autogenerated. Do not edit it by hand,
17
17
  since your modifications would be overwritten.
18
18
  """
19
19
 
20
+ from keras_hub.api import bounding_box
20
21
  from keras_hub.api import layers
21
22
  from keras_hub.api import metrics
22
23
  from keras_hub.api import models
23
24
  from keras_hub.api import samplers
24
25
  from keras_hub.api import tokenizers
26
+ from keras_hub.api import utils
25
27
  from keras_hub.src.utils.preset_utils import upload_preset
26
28
  from keras_hub.src.version_utils import __version__
27
29
  from keras_hub.src.version_utils import version
@@ -0,0 +1,36 @@
1
+ # Copyright 2024 The KerasHub Authors
2
+ #
3
+ # Licensed under the Apache License, Version 2.0 (the "License");
4
+ # you may not use this file except in compliance with the License.
5
+ # You may obtain a copy of the License at
6
+ #
7
+ # https://www.apache.org/licenses/LICENSE-2.0
8
+ #
9
+ # Unless required by applicable law or agreed to in writing, software
10
+ # distributed under the License is distributed on an "AS IS" BASIS,
11
+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ # See the License for the specific language governing permissions and
13
+ # limitations under the License.
14
+ """DO NOT EDIT.
15
+
16
+ This file was autogenerated. Do not edit it by hand,
17
+ since your modifications would be overwritten.
18
+ """
19
+
20
+ from keras_hub.src.bounding_box.converters import convert_format
21
+ from keras_hub.src.bounding_box.formats import CENTER_XYWH
22
+ from keras_hub.src.bounding_box.formats import REL_XYWH
23
+ from keras_hub.src.bounding_box.formats import REL_XYXY
24
+ from keras_hub.src.bounding_box.formats import REL_YXYX
25
+ from keras_hub.src.bounding_box.formats import XYWH
26
+ from keras_hub.src.bounding_box.formats import XYXY
27
+ from keras_hub.src.bounding_box.formats import YXYX
28
+ from keras_hub.src.bounding_box.iou import compute_ciou
29
+ from keras_hub.src.bounding_box.iou import compute_iou
30
+ from keras_hub.src.bounding_box.to_dense import to_dense
31
+ from keras_hub.src.bounding_box.to_ragged import to_ragged
32
+ from keras_hub.src.bounding_box.utils import as_relative
33
+ from keras_hub.src.bounding_box.utils import clip_boxes
34
+ from keras_hub.src.bounding_box.utils import clip_to_image
35
+ from keras_hub.src.bounding_box.utils import is_relative
36
+ from keras_hub.src.bounding_box.validate_format import validate_format
@@ -36,6 +36,8 @@ from keras_hub.src.layers.modeling.token_and_position_embedding import (
36
36
  )
37
37
  from keras_hub.src.layers.modeling.transformer_decoder import TransformerDecoder
38
38
  from keras_hub.src.layers.modeling.transformer_encoder import TransformerEncoder
39
+ from keras_hub.src.layers.preprocessing.audio_converter import AudioConverter
40
+ from keras_hub.src.layers.preprocessing.image_converter import ImageConverter
39
41
  from keras_hub.src.layers.preprocessing.masked_lm_mask_generator import (
40
42
  MaskedLMMaskGenerator,
41
43
  )
@@ -44,4 +46,16 @@ from keras_hub.src.layers.preprocessing.multi_segment_packer import (
44
46
  )
45
47
  from keras_hub.src.layers.preprocessing.random_deletion import RandomDeletion
46
48
  from keras_hub.src.layers.preprocessing.random_swap import RandomSwap
49
+ from keras_hub.src.layers.preprocessing.resizing_image_converter import (
50
+ ResizingImageConverter,
51
+ )
47
52
  from keras_hub.src.layers.preprocessing.start_end_packer import StartEndPacker
53
+ from keras_hub.src.models.pali_gemma.pali_gemma_image_converter import (
54
+ PaliGemmaImageConverter,
55
+ )
56
+ from keras_hub.src.models.resnet.resnet_image_converter import (
57
+ ResNetImageConverter,
58
+ )
59
+ from keras_hub.src.models.whisper.whisper_audio_converter import (
60
+ WhisperAudioConverter,
61
+ )