EYEDOL commited on
Commit
2e9bf08
·
verified ·
1 Parent(s): 0905af1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.")