Spaces:
Running
Running
remove revision
Browse files
app.py
CHANGED
|
@@ -67,8 +67,7 @@ def ensure_model_repo_on_pythonpath(model_name: str, revision: Optional[str] = N
|
|
| 67 |
init_hf_cache()
|
| 68 |
|
| 69 |
MODEL_NAME = os.getenv("MODEL_NAME", "stepfun-ai/Step3-VL-10B")
|
| 70 |
-
|
| 71 |
-
ensure_model_repo_on_pythonpath(MODEL_NAME, commit_hash)
|
| 72 |
|
| 73 |
_MODEL = None
|
| 74 |
_PROCESSOR = None
|
|
@@ -97,8 +96,6 @@ def get_model(model_name: Optional[str] = None):
|
|
| 97 |
if os.path.isdir(model_name):
|
| 98 |
# Ensure local custom modules (e.g., configuration_step_vl.py) are importable.
|
| 99 |
ensure_model_repo_on_pythonpath(model_name, revision)
|
| 100 |
-
else:
|
| 101 |
-
revision = commit_hash
|
| 102 |
_PROCESSOR = AutoProcessor.from_pretrained(
|
| 103 |
model_name,
|
| 104 |
trust_remote_code=True,
|
|
|
|
| 67 |
init_hf_cache()
|
| 68 |
|
| 69 |
MODEL_NAME = os.getenv("MODEL_NAME", "stepfun-ai/Step3-VL-10B")
|
| 70 |
+
ensure_model_repo_on_pythonpath(MODEL_NAME)
|
|
|
|
| 71 |
|
| 72 |
_MODEL = None
|
| 73 |
_PROCESSOR = None
|
|
|
|
| 96 |
if os.path.isdir(model_name):
|
| 97 |
# Ensure local custom modules (e.g., configuration_step_vl.py) are importable.
|
| 98 |
ensure_model_repo_on_pythonpath(model_name, revision)
|
|
|
|
|
|
|
| 99 |
_PROCESSOR = AutoProcessor.from_pretrained(
|
| 100 |
model_name,
|
| 101 |
trust_remote_code=True,
|