File tree Expand file tree Collapse file tree 2 files changed +20
-3
lines changed Expand file tree Collapse file tree 2 files changed +20
-3
lines changed Original file line number Diff line number Diff line change 1111# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1212# See the License for the specific language governing permissions and
1313# limitations under the License.
14+
15+ from .image_process_launcher import launcher as image_process
16+ from .modelscope_launcher import launcher as modelscope
17+ from .transformers_launcher import launcher as transformers
18+ from .yolo_launcher import launcher as yolo
Original file line number Diff line number Diff line change @@ -207,7 +207,11 @@ def test_cache_from_uri_local():
207207
208208
209209def test_custom_llm ():
210- from ..custom import get_user_defined_llm_families , register_llm , unregister_llm
210+ from ..custom import get_registered_llm_families as get_user_defined_llm_families
211+ from ..custom import (
212+ register_llm ,
213+ unregister_llm ,
214+ )
211215
212216 spec = LlamaCppLLMSpecV2 (
213217 model_format = "ggufv2" ,
@@ -239,7 +243,11 @@ def test_custom_llm():
239243
240244def test_persistent_custom_llm ():
241245 from ....constants import XINFERENCE_MODEL_DIR
242- from ..custom import get_user_defined_llm_families , register_llm , unregister_llm
246+ from ..custom import get_registered_llm_families as get_user_defined_llm_families
247+ from ..custom import (
248+ register_llm ,
249+ unregister_llm ,
250+ )
243251
244252 spec = LlamaCppLLMSpecV2 (
245253 model_format = "ggufv2" ,
@@ -663,7 +671,11 @@ def test_quert_engine_SGLang():
663671
664672
665673def test_query_engine_general ():
666- from ..custom import get_user_defined_llm_families , register_llm , unregister_llm
674+ from ..custom import get_registered_llm_families as get_user_defined_llm_families
675+ from ..custom import (
676+ register_llm ,
677+ unregister_llm ,
678+ )
667679 from ..llama_cpp .core import XllamaCppModel
668680 from ..llm_family import LLM_ENGINES , check_engine_by_spec_parameters
669681
You can’t perform that action at this time.
0 commit comments