Kangheng commited on
Commit
a73bdce
·
verified ·
1 Parent(s): 7422a74

remove revision

Browse files
Files changed (1) hide show
  1. app.py +1 -4
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
- commit_hash = os.getenv("COMMIT_HASH", "82465dc5c0bdd872d7849177e587345686551387")
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,