mindspore_gs.ptq.MFLlama2Helper

class mindspore_gs.ptq.MFLlama2Helper(config: MindFormerConfig = None)

Derived from 'NetworkHelper', a utility class for the MindFormers framework Llama2 network.

Parameters

config (MindFormerConfig) – MindFormerConfig for network.

Raises

TypeError – If input config is not an instance of MindFormerConfig.