MCPcopy Create free account
hub / github.com/OpenDriveLab/ReSim / split_model.py

File split_model.py

SwissArmyTransformer/examples/llama/split_model.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import torch
2from sat import AutoModel
3from transformers import LlamaTokenizer
4from sat.model.mixins import CachedAutoregressiveMixin

Callers

nothing calls this directly

Calls 5

CachedAutoregressiveMixinClass · 0.85
add_mixinMethod · 0.80
chatFunction · 0.70
printFunction · 0.50
from_pretrainedMethod · 0.45

Tested by

no test coverage detected