LukasHug commited on
Commit
020ea58
·
verified ·
1 Parent(s): 9ba58af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -516,7 +516,8 @@ Set the environment variable `model` to change the model:
516
  model_path = '/common-repos/LlavaGuard/models/LlavaGuard-v1.1-7b-full/smid_and_crawled_v2_with_augmented_policies/json-v16/llava'
517
 
518
  print(f"Loading model {model_path}")
519
- tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name, token=api_key)
 
520
  model.config.tokenizer_model_max_length = 2048 * 2
521
 
522
  exit_status = 0
 
516
  model_path = '/common-repos/LlavaGuard/models/LlavaGuard-v1.1-7b-full/smid_and_crawled_v2_with_augmented_policies/json-v16/llava'
517
 
518
  print(f"Loading model {model_path}")
519
+ # tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name, token=api_key)
520
+ load_selected_model(model_path)
521
  model.config.tokenizer_model_max_length = 2048 * 2
522
 
523
  exit_status = 0