refactor

Author: meichangsu1

src/twinkle/model/transformers/transformers.py

@@ -1,7 +1,7 @@
 # Copyright (c) ModelScope Contributors. All rights reserved.
 import contextlib
-import os
 import json
+import os
 import re
 from dataclasses import dataclass, field
 from typing import Dict, Any, List, Literal, Callable

src/twinkle/utils/transformers_utils.py

@@ -152,7 +152,7 @@ def get_llm_model(model, *, model_meta=None, inner_backbone: bool = True):
     except Exception:
         pass
 
-    # 2) Unwrap PEFT/Swift wrappers.
+    # 2) Unwrap PEFT wrappers.
     try:
         from peft import PeftModel  # type: ignore
 
@@ -161,14 +161,6 @@ def get_llm_model(model, *, model_meta=None, inner_backbone: bool = True):
     except Exception:
         pass
 
-    try:
-        from swift.tuners import SwiftModel  # type: ignore
-
-        if isinstance(model, SwiftModel):
-            model = model.model
-    except Exception:
-        pass
-
     # 3) Locate the language model module in multimodal containers via model_meta.
     if model_meta is None:
         model_meta = getattr(model, "model_meta", None)