Lumos update deb80ea
Da Yin commited on
How to use ai2lumos/lumos_unified_ground_iterative with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("question-answering", model="ai2lumos/lumos_unified_ground_iterative") # Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("ai2lumos/lumos_unified_ground_iterative")
model = AutoModelForCausalLM.from_pretrained("ai2lumos/lumos_unified_ground_iterative", device_map="auto")