michaelryoo
commited on
Update xgen-mm-vid-inference-script_hf.py
Browse files
xgen-mm-vid-inference-script_hf.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
from transformers import AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor, LogitsProcessor
|
2 |
import torch
|
3 |
|
4 |
-
model_name_or_path = "Salesforce/xgen-mm-vid-phi3-mini-r-v1.5-128tokens-
|
5 |
model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)
|
6 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, use_fast=False, legacy=False)
|
7 |
image_processor = AutoImageProcessor.from_pretrained(model_name_or_path, trust_remote_code=True)
|
|
|
1 |
from transformers import AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor, LogitsProcessor
|
2 |
import torch
|
3 |
|
4 |
+
model_name_or_path = "Salesforce/xgen-mm-vid-phi3-mini-r-v1.5-128tokens-8frames"
|
5 |
model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)
|
6 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, use_fast=False, legacy=False)
|
7 |
image_processor = AutoImageProcessor.from_pretrained(model_name_or_path, trust_remote_code=True)
|