Update services/vince_server.py

services/vince_server.py

@@ -7,12 +7,12 @@ from typing import List, Optional
 from omegaconf import OmegaConf, open_dict
 
 # Copiado/adaptado do vincie.py: uso de HF Hub
-from huggingface_hub import hf_hub_download, list_repo_files, HfApi  # [attached_file:2]
+from huggingface_hub import hf_hub_download, list_repo_files, HfApi
 
 VINCIE_DIR = Path(os.getenv("VINCIE_DIR", "/app/VINCIE"))
-VINCE_GIT_URL = os.getenv("VINCE_GIT_URL", "https://github.com/ByteDance-Seed/VINCIE")  # [attached_file:2]
-VINCE_REPO_ID = os.getenv("VINCE_REPO_ID", "ByteDance-Seed/VINCIE-3B")  # [attached_file:2]
-VINCE_CKPT = Path(os.getenv("VINCE_CKPT", "/app/ckpt/VINCIE-3B"))  # [attached_file:2]
+VINCE_GIT_URL = os.getenv("VINCE_GIT_URL", "https://github.com/ByteDance-Seed/VINCIE")
+VINCE_REPO_ID = os.getenv("VINCE_REPO_ID", "ByteDance-Seed/VINCIE-3B")
+VINCE_CKPT = Path(os.getenv("VINCE_CKPT", "/app/ckpt/VINCIE-3B"))
 
 if str(VINCIE_DIR) not in sys.path:
     sys.path.insert(0, str(VINCIE_DIR))
@@ -33,9 +33,10 @@ class VinceServer:
         self.config_path = config_path
         self.output_root = Path(output_root); self.output_root.mkdir(parents=True, exist_ok=True)
         overrides = list(base_overrides or [])
+
         self.HF_HOME = Path(os.getenv("HF_HOME", "/data/.cache/huggingface"))
         self.HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN") or None
-        
+       
         # 1) Clonar/garantir repo
         self.ensure_repo()
 
@@ -63,7 +64,7 @@ class VinceServer:
     # ==== Helpers de bootstrap copiados/adaptados do vincie.py ====
 
     def ensure_repo(self) -> None:
-        """Clona o repositório oficial quando ausente (idempotente)."""  # [attached_file:2]
+        """Clona o repositório oficial quando ausente (idempotente)."""
         if not VINCIE_DIR.exists():
             subprocess.run(["git", "clone", VINCE_GIT_URL, str(VINCIE_DIR)], check=True)
 
@@ -71,7 +72,7 @@ class VinceServer:
         """
         Baixa TODOS os arquivos do modelo do Hub (VINCE_REPO_ID) para VINCE_CKPT,
         preservando subdiretórios. Critério de pular download: arquivo > 1MB.
-        Cria pastas necessárias.  # [attached_file:2]
+        Cria pastas necessárias.
         """
         VINCE_CKPT.mkdir(parents=True, exist_ok=True)
         token = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_TOKEN")
@@ -106,12 +107,13 @@ class VinceServer:
                     hf_hub_download(
                         repo_id=VINCE_REPO_ID,
                         filename=rel_path,
-                        local_dir=str(VINCE_CKPT),
                         cache_dir=str(self.HF_HOME),
+                        local_dir=str(VINCE_CKPT),
                         #local_dir_use_symlinks=False,
                         token=token,
                         #force_download=False,
                         #local_files_only=False,
+                        #revision=repo_revision,
                     )
                     downloaded += 1
                 except Exception as de:
@@ -122,7 +124,7 @@ class VinceServer:
         """
         Cria symlink compatível para caminhos relativos do repo:
         /app/VINCIE/ckpt/VINCIE-3B -> /app/ckpt/VINCIE-3B.
-        Não remove diretórios reais; apenas substitui symlink divergente.  # [attached_file:2]
+        Não remove diretórios reais; apenas substitui symlink divergente.
         """
         repo_ckpt_dir = VINCIE_DIR / "ckpt"
         repo_ckpt_dir.mkdir(parents=True, exist_ok=True)
@@ -166,8 +168,6 @@ class VinceServer:
                 raise RuntimeError(f"[vince_server] missing bootstrap step: {name}")
             fn()
 
-        
-
     def _apply_generation_overrides(self, cfg, *, output_dir: Path,
                                     image_path: Optional[str]=None,
                                     prompts: Optional[List[str]]=None,
@@ -213,7 +213,13 @@ class VinceServer:
             fn = getattr(self.gen, name, None)
             if callable(fn):
                 print(f"[vince_server] using inference method: {name}")
-                return fn(self.config)
+                # =============================================================
+                # CORREÇÃO APLICADA AQUI
+                # O método de inferência (ex: inference_loop) não espera
+                # argumentos, pois acessa a configuração através do `self`
+                # do seu próprio objeto (`self.gen`).
+                return fn()
+                # =============================================================
         raise RuntimeError("[vince_server] no valid inference method found")
 
     def _cleanup(self):
@@ -266,4 +272,4 @@ class VinceServer:
         self.config = cfg
         result = self._infer_once()
         self._cleanup()
-        return out_dir
+        return out_dir