Types & classes — github.com/CERT-Lab/lora-sb

CodeHub MCPcopy Create free account

hub / github.com/CERT-Lab/lora-sb / types & classes

Types & classes9 in github.com/CERT-Lab/lora-sb

⨍ Functions 134◇ Types & classes 9

↓ 1 callersClassDataCollatorForSupervisedDataset

Collate examples for supervised fine-tuning.

utils/data_utils.py:153

↓ 1 callersClassForwardBackwardOffloadHookContext

utils/offload_utils_for_quant/forward_backward_offload.py:7

↓ 1 callersClassGradientOffloadHookContext

utils/offload_utils_for_quant/gradient_offload.py:4

↓ 1 callersClassModelOffloadHookContext

utils/offload_utils_for_quant/model_offload.py:5

↓ 1 callersClassOffloadContext

utils/offload_utils_for_quant/context.py:8

↓ 1 callersClassSavedTensorOffloadContext

utils/offload_utils_for_quant/saved_tensor_offload.py:5

ClassForwardHookForDevice

utils/offload_utils_for_quant/forward_hook.py:4

instruction_tuning_eval/utils.py:259

ClassOffloadSavedTensorHook

utils/offload_utils_for_quant/graph_hook.py:14