Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ print(f"✅ Finished encoding {len(texts_all)} texts. Shape: {text_embeds_all.sh
|
|
| 58 |
# Startup: load Llava model & tokenizer
|
| 59 |
# -------------------------
|
| 60 |
print("⏳ Loading Llava model and tokenizer...")
|
| 61 |
-
llava_tokenizer = AutoTokenizer.from_pretrained(LLAVA_MODEL_ID)
|
| 62 |
llava_model = AutoModelForCausalLM.from_pretrained(LLAVA_MODEL_ID).to(device)
|
| 63 |
llava_model.eval()
|
| 64 |
print("✅ Llava model loaded.")
|
|
|
|
| 58 |
# Startup: load Llava model & tokenizer
|
| 59 |
# -------------------------
|
| 60 |
print("⏳ Loading Llava model and tokenizer...")
|
| 61 |
+
llava_tokenizer = AutoTokenizer.from_pretrained(LLAVA_MODEL_ID, use_fast=False)
|
| 62 |
llava_model = AutoModelForCausalLM.from_pretrained(LLAVA_MODEL_ID).to(device)
|
| 63 |
llava_model.eval()
|
| 64 |
print("✅ Llava model loaded.")
|