bhsinghgrid commited on
Commit
b057b70
·
verified ·
1 Parent(s): b5adaa5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +18 -9
app.py CHANGED
@@ -325,18 +325,27 @@ def build_runtime_cfg(ckpt_path: str, item: dict | None = None):
325
  hf_settings = {}
326
  cfg = copy.deepcopy(CONFIG)
327
  if root == "hf":
328
- cfg["model_type"] = os.environ.get(
329
- "HF_DEFAULT_MODEL_TYPE",
330
- hf_settings.get("model_type", HF_DEFAULT_SETTINGS.get("model_type", HF_DEFAULT_MODEL_TYPE)),
 
 
331
  )
332
- include_neg_raw = os.environ.get(
333
- "HF_DEFAULT_INCLUDE_NEG",
334
- str(hf_settings.get("include_negative_examples", HF_DEFAULT_SETTINGS.get("include_negative_examples", HF_DEFAULT_INCLUDE_NEG))),
 
 
 
 
 
335
  )
336
  cfg["data"]["include_negative_examples"] = include_neg_raw.lower() == "true"
337
- t_raw = os.environ.get(
338
- "HF_DEFAULT_NUM_STEPS",
339
- hf_settings.get("num_steps", HF_DEFAULT_SETTINGS.get("num_steps", HF_DEFAULT_NUM_STEPS)),
 
 
340
  )
341
  if t_raw:
342
  t_val = int(t_raw)
 
325
  hf_settings = {}
326
  cfg = copy.deepcopy(CONFIG)
327
  if root == "hf":
328
+ cfg["model_type"] = (
329
+ hf_settings.get("model_type")
330
+ or os.environ.get("HF_DEFAULT_MODEL_TYPE")
331
+ or HF_DEFAULT_SETTINGS.get("model_type")
332
+ or HF_DEFAULT_MODEL_TYPE
333
  )
334
+ include_neg_raw = str(
335
+ hf_settings.get(
336
+ "include_negative_examples",
337
+ os.environ.get(
338
+ "HF_DEFAULT_INCLUDE_NEG",
339
+ HF_DEFAULT_SETTINGS.get("include_negative_examples", HF_DEFAULT_INCLUDE_NEG),
340
+ ),
341
+ )
342
  )
343
  cfg["data"]["include_negative_examples"] = include_neg_raw.lower() == "true"
344
+ t_raw = (
345
+ hf_settings.get("num_steps")
346
+ or os.environ.get("HF_DEFAULT_NUM_STEPS")
347
+ or HF_DEFAULT_SETTINGS.get("num_steps")
348
+ or HF_DEFAULT_NUM_STEPS
349
  )
350
  if t_raw:
351
  t_val = int(t_raw)