LukasHug commited on
Commit
0e8e2ae
1 Parent(s): ac4a50f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -54,7 +54,7 @@ from transformers import AutoProcessor, LlavaForConditionalGeneration
54
  from PIL import Image
55
  import requests
56
 
57
- model = LlavaForConditionalGeneration.from_pretrained('AIML-TUDA/LlavaGuard-v1.2-7B-OV-hf')
58
  processor = AutoProcessor.from_pretrained('AIML-TUDA/LlavaGuard-v1.2-7B-OV-hf')
59
 
60
  conversation = [
 
54
  from PIL import Image
55
  import requests
56
 
57
+ model = LlavaOnevisionForConditionalGeneration.from_pretrained('AIML-TUDA/LlavaGuard-v1.2-7B-OV-hf')
58
  processor = AutoProcessor.from_pretrained('AIML-TUDA/LlavaGuard-v1.2-7B-OV-hf')
59
 
60
  conversation = [