diff --git a/explainability/explainer/base_explainer.py b/explainability/explainer/base_explainer.py index dc4fc62..bb81510 100644 --- a/explainability/explainer/base_explainer.py +++ b/explainability/explainer/base_explainer.py @@ -18,6 +18,7 @@ def __init__( branch_level: Literal["low", "high"] = "high", l_stage_idx: int = -1, # gcvit low branch stage index (0~3) block_idx: int = -1, # vit, gcvit block index + model: Optional[torch.nn.Module] = None, ): self.device = "cuda:0" if torch.cuda.is_available() else 'cpu' @@ -34,7 +35,9 @@ def __init__( self.transformer_type = model_name.split("_")[-2] # vit, gcvit # Model & Tools setup - self.model = timm.create_model(model_name, pretrained=True, dataset=dataset) + self.model = model if model is not None else timm.create_model( + model_name, pretrained=True, dataset=dataset + ) self.face_detector = FaceDetector2(conf_thres) self.img_size = [224,224] if self.model_variant == "b0" else [384,384] self.transforms = get_test_transforms(img_size=self.img_size) diff --git a/pyproject.toml b/pyproject.toml index 1361336..5f7357f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -8,7 +8,7 @@ build-backend = "setuptools.build_meta" # ───────────────────────────────────────────── [project] name = "deepguard" # pip install deepguard 할 때 이 이름 -version = "0.2.4" # pip install deepguard==0.2.4 으로 버전 고정 가능 +version = "0.3.0" # pip install deepguard==0.2.4 으로 버전 고정 가능 description = "Multi-Scale Efficient Global Context Vision Transformer for Robust Deepfake Detection" readme = "README.md" # PyPI 페이지에 표시될 설명 (마크다운) requires-python = ">=3.10" # 이 버전 미만이면 pip install 자체를 막음