Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|