simplify code

TUDB-Labs · Jan 5, 2024 · 6998658 · 6998658
1 parent def320e
commit 6998658
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/mlora.py b/mlora.py
@@ -95,10 +95,8 @@ def init_adapter_config(config: Dict[str, any],
                         llm_model: mlora.LLMModel,
                         ) -> List[Union[mlora.GenerateConfig, mlora.TrainConfig]]:
     if args.disable_adapter and args.inference:
-        config_class = mlora.LoraConfig(
-            adapter_name_="m-LoRA", device_=args.device)
         config_class = mlora.GenerateConfig().init(
-            config_class)
+            mlora.LoraConfig(adapter_name_="m-LoRA", device_=args.device))
         if args.prompt_template:
             config_class.prompt_template_ = "template/template_demo.json"
         return {"m-LoRA": config_class}