File size: 1,764 Bytes
d0ffe9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
try:
from ._version import (
version as __version__,
version_tuple,
)
except ImportError:
__version__ = "unknown (no version information available)"
version_tuple = (0, 0, "unknown", "noinfo")
from functools import lru_cache
from os import getenv
from pathlib import Path
from warnings import filterwarnings
from rich.console import Console
from tqdm import TqdmExperimentalWarning
PACKAGE = __package__.replace("_", "-")
PACKAGE_ROOT = Path(__file__).parent.parent
HF_HOME = Path(getenv("HF_HOME", Path.home() / ".cache" / "huggingface"))
HF_HUB_CACHE = Path(getenv("HUGGINGFACE_HUB_CACHE", HF_HOME.joinpath("hub")))
HF_LIB_NAME = "animatediff-cli"
HF_LIB_VER = __version__
HF_MODULE_REPO = "neggles/animatediff-modules"
console = Console(highlight=True)
err_console = Console(stderr=True)
# shhh torch, don't worry about it it's fine
filterwarnings("ignore", category=UserWarning, message="TypedStorage is deprecated")
# you too tqdm
filterwarnings("ignore", category=TqdmExperimentalWarning)
@lru_cache(maxsize=4)
def get_dir(dirname: str = "data") -> Path:
if PACKAGE_ROOT.name == "src":
# we're installed in editable mode from within the repo
dirpath = PACKAGE_ROOT.parent.joinpath(dirname)
else:
# we're installed normally, so we just use the current working directory
dirpath = Path.cwd().joinpath(dirname)
dirpath.mkdir(parents=True, exist_ok=True)
return dirpath.absolute()
__all__ = [
"__version__",
"version_tuple",
"PACKAGE",
"PACKAGE_ROOT",
"HF_HOME",
"HF_HUB_CACHE",
"console",
"err_console",
"get_dir",
"models",
"pipelines",
"rife",
"utils",
"cli",
"generate",
"schedulers",
"settings",
]
|