Spaces:
Runtime error
Runtime error
File size: 431 Bytes
413d4d0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import os
from ._version import __version__
MODEL_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", "checkpoints"))
if os.environ.get("VIDEO_MODEL_PATH"):
MODEL_PATH = os.environ.get("VIDEO_MODEL_PATH")
# (cogVideo) Set the SAT_HOME env variable to MODEL_PATH if not set
if not os.environ.get("SAT_HOME"):
os.environ["SAT_HOME"] = MODEL_PATH
from .infermodels import load, get_model, load_model
|