diff --git a/__init__.py b/__init__.py index 79f3f46..ac0839e 100644 --- a/__init__.py +++ b/__init__.py @@ -19,7 +19,7 @@ from spar3d.utils import foreground_crop SPAR3D_CATEGORY = "SPAR3D" -SPAR3D_MODEL_NAME = "stabilityai/spar3d" +SPAR3D_MODEL_NAME = os.getenv("SPAR3D_MODEL_NAME", "stabilityai/stable-fast-3d") class SPAR3DLoader: