InternVL3-78B / app.py
zhuangjia777's picture
Create app.py
e0a8e8f verified
raw
history blame contribute delete
267 Bytes
import torch
from transformers import AutoTokenizer, AutoModel
path = "OpenGVLab/InternVL3-78B"
model = AutoModel.from_pretrained(
path,
torch_dtype=torch.bfloat16,
low_cpu_mem_usage=True,
use_flash_attn=True,
trust_remote_code=True).eval().cuda()