Update requirements

app.py

@@ -14,7 +14,7 @@ model_name = 'AIDC-AI/Ovis1.6-Llama3.2-3B'
 model = AutoModelForCausalLM.from_pretrained(model_name,
                                              torch_dtype=torch.bfloat16,
                                              multimodal_max_length=8192,
-                                             use_auth_token=os.getenv('HUGGINGFACE_TOKEN'),
+                                             token=os.getenv('HUGGINGFACE_TOKEN'),
                                              trust_remote_code=True).to(device='cuda')
 text_tokenizer = model.get_text_tokenizer()
 visual_tokenizer = model.get_visual_tokenizer()

requirements.txt

@@ -1,3 +1,4 @@
 numpy==1.24.3
 torch==2.2.0
 transformers==4.44.2
+flash-attn==2.6.3