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

File transform_param.py

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

Source from the content-addressed store, hash-verified

1from transformers import LlamaForCausalLM, LlamaTokenizer, LlamaConfig
2
3prefix = 'meta-llama/'
4model_type = 'Llama-2-70b-chat-hf'

Callers

nothing calls this directly

Calls 5

LLaMAModelClass · 0.90
save_checkpointFunction · 0.90
get_mixinMethod · 0.80
transform_weightFunction · 0.70
from_pretrainedMethod · 0.45

Tested by

no test coverage detected