Spaces:
Runtime error
Runtime error
File size: 267 Bytes
e0a8e8f |
1 2 3 4 5 6 7 8 9 |
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() |