SaulSantos commited on
Commit
fca5de9
·
verified ·
1 Parent(s): 6ea1b9e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -75,12 +75,12 @@ from transformers import AutoProcessor, LlavaOnevisionForConditionalGeneration
75
 
76
  # Load the model in half-precision
77
  model = LlavaOnevisionForConditionalGeneration.from_pretrained(
78
- "/mnt/data-poseidon/saul/towerpvideo_hf",
79
  device_map="auto"
80
  )
81
 
82
  processor = AutoProcessor.from_pretrained(
83
- "utter-project/TowerVideo-7B"
84
  )
85
 
86
  # Use your local video
 
75
 
76
  # Load the model in half-precision
77
  model = LlavaOnevisionForConditionalGeneration.from_pretrained(
78
+ "utter-project/TowerVideo-2B",
79
  device_map="auto"
80
  )
81
 
82
  processor = AutoProcessor.from_pretrained(
83
+ "utter-project/TowerVideo-2B"
84
  )
85
 
86
  # Use your local video