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

File chat_sat.py

SwissArmyTransformer/examples/mixtral-8x7b/chat_sat.py:None–None  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1import torch
2import argparse
3from sat.model.mixins import CachedAutoregressiveMixin
4from sat.quantization.kernels import quantize

Callers

nothing calls this directly

Calls 15

get_model_parallel_world_sizeFunction · 0.90
quantizeFunction · 0.90
stream_filling_sequenceFunction · 0.90
get_model_parallel_rankFunction · 0.90
filling_sequenceFunction · 0.90
CachedAutoregressiveMixinClass · 0.85
chat_history_to_promptFunction · 0.85
BaseStrategyClass · 0.85
getMethod · 0.80
add_mixinMethod · 0.80
appendMethod · 0.80
printFunction · 0.50

Tested by

no test coverage detected