Update hf_model.py

hf_model.py

@@ -70,18 +70,6 @@ class UniBioTransferModel(LatentDiffusion, PyTorchModelHubMixin):
         **kwargs: Additional arguments passed to LatentDiffusion
     """
 
-    def __getstate__(self):
-        state = self.__dict__.copy()
-        if 'ptsM_Generator' in state:
-            del state['ptsM_Generator']
-        return state
-
-    def __getattr__(self, name):
-        if name == 'ptsM_Generator':
-            self.ptsM_Generator = LandmarkExtractor(include_visualizer=True, img_256_mode=False)
-            return self.ptsM_Generator
-        raise AttributeError(f"'{type(self).__name__}' object has no attribute '{name}'")
-
     def __init__(self, config=None, task="face", **kwargs):
         self._task_name = task if isinstance(task, str) else TASK_ID2NAME.get(task, "face")
         self._task_id = TASK_NAME2ID.get(self._task_name, 0) if isinstance(task, str) else task
@@ -241,7 +229,6 @@ class UniBioTransferModel(LatentDiffusion, PyTorchModelHubMixin):
 
         # offload_unused_tasks__LD(model, task_id, method="cpu")
         
-        model.ptsM_Generator = LandmarkExtractor(include_visualizer=True, img_256_mode=False)
         cleanup_gpu_memory()
         
         # ZeroGPU 兼容：只在 device 不是 "cpu" 且 CUDA 可用时才移动到 GPU