mirror of
				https://github.com/hiyouga/LLaMA-Factory.git
				synced 2025-11-04 18:02:19 +08:00 
			
		
		
		
	[misc] lint (#9221)
This commit is contained in:
		
							parent
							
								
									0c17b55ddc
								
							
						
					
					
						commit
						2a797e2562
					
				@ -49,10 +49,12 @@ class ChatModel:
 | 
			
		||||
 | 
			
		||||
        if model_args.infer_backend == EngineName.HF:
 | 
			
		||||
            from .hf_engine import HuggingfaceEngine
 | 
			
		||||
 | 
			
		||||
            self.engine: BaseEngine = HuggingfaceEngine(model_args, data_args, finetuning_args, generating_args)
 | 
			
		||||
        elif model_args.infer_backend == EngineName.VLLM:
 | 
			
		||||
            try:
 | 
			
		||||
                from .vllm_engine import VllmEngine
 | 
			
		||||
 | 
			
		||||
                self.engine: BaseEngine = VllmEngine(model_args, data_args, finetuning_args, generating_args)
 | 
			
		||||
            except ImportError as e:
 | 
			
		||||
                raise ImportError(
 | 
			
		||||
@ -62,6 +64,7 @@ class ChatModel:
 | 
			
		||||
        elif model_args.infer_backend == EngineName.SGLANG:
 | 
			
		||||
            try:
 | 
			
		||||
                from .sglang_engine import SGLangEngine
 | 
			
		||||
 | 
			
		||||
                self.engine: BaseEngine = SGLangEngine(model_args, data_args, finetuning_args, generating_args)
 | 
			
		||||
            except ImportError as e:
 | 
			
		||||
                raise ImportError(
 | 
			
		||||
 | 
			
		||||
@ -37,36 +37,43 @@ USAGE = (
 | 
			
		||||
 | 
			
		||||
def _run_api():
 | 
			
		||||
    from .api.app import run_api
 | 
			
		||||
 | 
			
		||||
    return run_api()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _run_chat():
 | 
			
		||||
    from .chat.chat_model import run_chat
 | 
			
		||||
 | 
			
		||||
    return run_chat()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _run_eval():
 | 
			
		||||
    from .eval.evaluator import run_eval
 | 
			
		||||
 | 
			
		||||
    return run_eval()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _export_model():
 | 
			
		||||
    from .train.tuner import export_model
 | 
			
		||||
 | 
			
		||||
    return export_model()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _run_exp():
 | 
			
		||||
    from .train.tuner import run_exp
 | 
			
		||||
 | 
			
		||||
    return run_exp()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _run_web_demo():
 | 
			
		||||
    from .webui.interface import run_web_demo
 | 
			
		||||
 | 
			
		||||
    return run_web_demo()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _run_web_ui():
 | 
			
		||||
    from .webui.interface import run_web_ui
 | 
			
		||||
 | 
			
		||||
    return run_web_ui()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user