michaelryoo commited on
Commit
b072331
·
verified ·
1 Parent(s): 77067ce

Update xgen-mm-vid-inference-script_hf.py

Browse files
Files changed (1) hide show
  1. xgen-mm-vid-inference-script_hf.py +1 -1
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-16frames"
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)