SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update llava.md (#34749)

Browse Source 
LLava -> Llava
This commit is contained in:
Lysandre Debut
2024-11-15 15:39:57 +01:00
committed by GitHub
parent 8ba3e1505e
commit f5dbfab7f3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/en/model_doc/llava.md
View File

@@ -85,10 +85,10 @@ LLaVa also supports batched inference. Here is how you can do it:
import requests import requests
from PIL import Image from PIL import Image
import torch import torch
from transformers import AutoProcessor, LLavaForConditionalGeneration from transformers import AutoProcessor, LlavaForConditionalGeneration
# Load the model in half-precision # Load the model in half-precision
model = LLavaForConditionalGeneration.from_pretrained("llava-hf/llava-1.5-7b-hf", torch_dtype=torch.float16, device_map="auto") model = LlavaForConditionalGeneration.from_pretrained("llava-hf/llava-1.5-7b-hf", torch_dtype=torch.float16, device_map="auto")
processor = AutoProcessor.from_pretrained("llava-hf/llava-1.5-7b-hf") processor = AutoProcessor.from_pretrained("llava-hf/llava-1.5-7b-hf")
# Get two different images # Get two different images