meta.py 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157
  1. # copyright (c) 2024 PaddlePaddle Authors. All Rights Reserve.
  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. # http://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. __all__ = ["get_all_repo_names"]
  15. REPO_DOWNLOAD_BASE = (
  16. "https://paddle-model-ecology.bj.bcebos.com/paddlex/PaddleX3.0/repos/"
  17. )
  18. REPO_NAMES = [
  19. "PaddleClas",
  20. "PaddleOCR",
  21. "PaddleDetection",
  22. "PaddleSeg",
  23. "PaddleNLP",
  24. "PaddleTS",
  25. "PaddleVideo",
  26. ]
  27. REPO_META = {
  28. "PaddleSeg": {
  29. "git_path": "/PaddlePaddle/PaddleSeg.git",
  30. "platform": "github",
  31. "branch": "develop",
  32. "pkg_name": "paddleseg",
  33. "lib_name": "paddleseg",
  34. "pdx_pkg_name": "PaddleSeg_api",
  35. "editable": False,
  36. "extra_req_files": ["Matting/requirements.txt"],
  37. "path_env": "PADDLE_PDX_PADDLESEG_PATH",
  38. },
  39. "PaddleClas": {
  40. "git_path": "/PaddlePaddle/PaddleClas.git",
  41. "platform": "github",
  42. "branch": "develop",
  43. "pkg_name": "paddleclas",
  44. "lib_name": "paddleclas",
  45. "pdx_pkg_name": "PaddleClas_api",
  46. # PaddleClas must be installed in non-editable mode, otherwise it throws
  47. # an Import error.
  48. "editable": False,
  49. "path_env": "PADDLE_PDX_PADDLECLAS_PATH",
  50. },
  51. "PaddleDetection": {
  52. "git_path": "/PaddlePaddle/PaddleDetection.git",
  53. "platform": "github",
  54. "branch": "develop",
  55. "pkg_name": "paddledet",
  56. "lib_name": "ppdet",
  57. "pdx_pkg_name": "PaddleDetection_api",
  58. "editable": False,
  59. "path_env": "PADDLE_PDX_PADDLEDETECTION_PATH",
  60. },
  61. "PaddleOCR": {
  62. "git_path": "/PaddlePaddle/PaddleOCR.git",
  63. "platform": "github",
  64. "branch": "main",
  65. "pkg_name": "paddleocr",
  66. "lib_name": "paddleocr",
  67. "pdx_pkg_name": "PaddleOCR_api",
  68. "editable": False,
  69. "extra_req_files": [
  70. "ppstructure/kie/requirements.txt",
  71. "docs/algorithm/formula_recognition/requirements.txt",
  72. ],
  73. "path_env": "PADDLE_PDX_PADDLEOCR_PATH",
  74. "requires": ["PaddleNLP"],
  75. },
  76. "PaddleTS": {
  77. "git_path": "/PaddlePaddle/PaddleTS.git",
  78. "platform": "github",
  79. "branch": "release_v1.1",
  80. "pkg_name": "paddlets",
  81. "lib_name": "paddlets",
  82. "pdx_pkg_name": "PaddleTS_api",
  83. "editable": False,
  84. "path_env": "PADDLE_PDX_PADDLETS_PATH",
  85. "pdx_pkg_deps": ["pandas", "ruamel.yaml"],
  86. },
  87. "PaddleNLP": {
  88. "git_path": "/PaddlePaddle/PaddleNLP.git",
  89. "platform": "github",
  90. "branch": "release/2.9",
  91. "pkg_name": "paddlenlp",
  92. "lib_name": "paddlenlp",
  93. "pdx_pkg_name": "PaddleNLP_api",
  94. "editable": False,
  95. "path_env": "PADDLE_PDX_PADDLENLP_PATH",
  96. },
  97. "PaddleSpeech": {
  98. "git_path": "/PaddlePaddle/PaddleSpeech.git",
  99. "platform": "github",
  100. "branch": "develop",
  101. "pkg_name": "paddlespeech",
  102. "lib_name": "paddlespeech",
  103. "pdx_pkg_name": "PaddleSpeech_api",
  104. "editable": False,
  105. "path_env": "PADDLE_PDX_PADDLESPEECH_PATH",
  106. "requires": ["PaddleNLP"],
  107. },
  108. "PARL": {
  109. "git_path": "/PaddlePaddle/PARL.git",
  110. "platform": "github",
  111. "branch": "develop",
  112. "pkg_name": "parl",
  113. "lib_name": "parl",
  114. "pdx_pkg_name": "PARL_api",
  115. "editable": False,
  116. "path_env": "PADDLE_PDX_PARL_PATH",
  117. },
  118. "PaddleMIX": {
  119. "git_path": "/PaddlePaddle/PaddleMIX.git",
  120. "platform": "github",
  121. "branch": "develop",
  122. "pkg_name": "paddlemix",
  123. "lib_name": "paddlemix",
  124. "pdx_pkg_name": "PaddleMIX_api",
  125. "editable": True,
  126. "extra_editable": "ppdiffusers",
  127. "path_env": "PADDLE_PDX_PADDLEMIX_PATH",
  128. "requires": ["PaddleNLP"],
  129. },
  130. "PaddleVideo": {
  131. "git_path": "/PaddlePaddle/PaddleVideo.git",
  132. "platform": "github",
  133. "branch": "develop",
  134. "pkg_name": "paddlevideo",
  135. "lib_name": "ppvideo",
  136. "pdx_pkg_name": "PaddleVideo_api",
  137. "editable": False,
  138. "extra_req_files": [
  139. "requirements_paddlex.txt",
  140. ],
  141. "path_env": "PADDLE_PDX_PADDLEVIDEO_PATH",
  142. },
  143. }
  144. def get_repo_meta(repo_name):
  145. """get_repo_meta"""
  146. return REPO_META[repo_name]
  147. def get_all_repo_names():
  148. """get_all_repo_names"""
  149. return REPO_NAMES