73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
import sys
|
|
import os
|
|
import logging
|
|
import subprocess
|
|
import traceback
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
log = logging.getLogger("AniDoc")
|
|
|
|
download_models = True
|
|
|
|
try:
|
|
import folder_paths
|
|
DIFFUSERS_DIR = os.path.join(folder_paths.models_dir, "diffusers")
|
|
ANIDOC_DIR = os.path.join(DIFFUSERS_DIR, "anidoc")
|
|
SVD_I2V_DIR = os.path.join(
|
|
DIFFUSERS_DIR,
|
|
"stable-video-diffusion-img2vid-xt",
|
|
)
|
|
except:
|
|
download_models = False
|
|
log.info("Not called by ComfyUI Manager. Models will not be downloaded")
|
|
|
|
EXT_PATH = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
COTRACKER = os.path.join(EXT_PATH, "cotracker")
|
|
|
|
try:
|
|
log.info(f"Installing requirements")
|
|
subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", f"{EXT_PATH}/requirements.txt", "--no-warn-script-location"])
|
|
|
|
if download_models:
|
|
from huggingface_hub import snapshot_download
|
|
|
|
try:
|
|
if not os.path.exists(ANIDOC_DIR):
|
|
log.info(f"Downloading AniDoc model to: {ANIDOC_DIR}")
|
|
snapshot_download(
|
|
repo_id="Yhmeng1106/anidoc",
|
|
local_dir=DIFFUSERS_DIR,
|
|
local_dir_use_symlinks=False,
|
|
)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
log.error(f"Failed to download AniDoc model")
|
|
|
|
try:
|
|
if not os.path.exists(SVD_I2V_DIR):
|
|
log.info(f"Downloading stable diffusion video img2vid to: {SVD_I2V_DIR}")
|
|
snapshot_download(
|
|
repo_id="vdo/stable-video-diffusion-img2vid-xt-1-1",
|
|
allow_patterns=[f"*.json", "*fp16*"],
|
|
ignore_patterns=["*unet*"],
|
|
local_dir=SVD_I2V_DIR,
|
|
local_dir_use_symlinks=False,
|
|
)
|
|
except Exception:
|
|
traceback.print_exc()
|
|
log.error(f"Failed to download stable diffusion video img2vid")
|
|
|
|
try:
|
|
log.info("Installing CoTracker")
|
|
subprocess.check_call([sys.executable, "-m", "pip", "install", COTRACKER])
|
|
except Exception:
|
|
traceback.print_exc()
|
|
log.error(f"Failed to install CoTracker")
|
|
|
|
log.info(f"AniDoc Installation completed")
|
|
|
|
except Exception:
|
|
traceback.print_exc()
|
|
log.error(f"AniDoc Installation failed") |