Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/CERT-Lab/lora-sb
/ functions
Functions
134 in github.com/CERT-Lab/lora-sb
⨍
Functions
134
◇
Types & classes
9
Method
after_hook
(module: torch.nn.Module, args, output)
utils/offload_utils_for_quant/forward_hook.py:65
Method
after_hook
(module, grad_input, grad_output)
utils/offload_utils_for_quant/forward_backward_offload.py:178
Method
after_hook_with_kwargs
(module, args, kwargs, output)
utils/offload_utils_for_quant/forward_hook.py:78
Method
after_hook_with_kwargs
(module, args, kwargs, output)
utils/offload_utils_for_quant/forward_backward_offload.py:146
Function
clean_numbers
(sample)
instruction_tuning_eval/utils.py:57
Function
cleanup_wandb
()
train_glue.py:53
Function
compute_metrics
(eval_preds)
utils/compute_utils.py:18
Function
create_compute_metrics_function
Create a compute_metrics function with the required parameters.
utils/compute_utils.py:3
Function
create_model_tokenizer_cr
(args)
models.py:92
Function
forward_latent
(self, x: torch.Tensor)
utils/latent_utils.py:53
Function
generate_and_tokenize_prompt_wrapper
(data_point)
utils/data_utils.py:246
Function
get_delta_weight
Compute the delta weight for the given adapter. Args: adapter (str): The name of the adapter for which the delta weight
utils/latent_utils.py:10
Method
get_full_name_list
Get the module name list of the leaf nodes of the module tree
utils/offload_utils_for_quant/forward_hook.py:95
Function
get_lr_scheduler
(optimizer, num_warmup_steps, num_training_steps)
train_eval.py:139
Function
get_model_memory
Calculates the estimated memory usage of a model in gigabytes. Args: model (torch.nn.Module): The model whose memory usage is to be
utils/offload_utils_for_quant/split.py:4
Method
hook
(module: torch.nn.Module, args)
utils/offload_utils_for_quant/forward_hook.py:14
Method
hook_with_kwargs
(module: torch.nn.Module, args, kwargs)
utils/offload_utils_for_quant/forward_hook.py:25
Function
is_equiv
(str1, str2, verbose=False)
instruction_tuning_eval/utils.py:237
Function
last_boxed_only
(sample)
instruction_tuning_eval/utils.py:5
Function
math_equal_process
(param)
instruction_tuning_eval/grader.py:98
Function
only_until_first_boxed_from_tokens
(string, tokens)
instruction_tuning_eval/utils.py:41
Method
pack
(x: torch.Tensor)
utils/offload_utils_for_quant/graph_hook.py:24
Method
pre_hook
(module: torch.nn.Module, args)
utils/offload_utils_for_quant/forward_hook.py:60
Method
pre_hook
(module, grad_output)
utils/offload_utils_for_quant/forward_backward_offload.py:174
Method
pre_hook_with_kwargs
(module, args, kwargs)
utils/offload_utils_for_quant/forward_hook.py:72
Method
pre_hook_with_kwargs
(module, args, kwargs)
utils/offload_utils_for_quant/forward_backward_offload.py:126
Function
record_gradient_hook
(grad)
utils/gradient_utils.py:23
Method
record_gradient_hook
(grad)
utils/offload_utils_for_quant/gradient_offload.py:46
Function
symbolic_equal_process
(a, b, output_queue)
instruction_tuning_eval/grader.py:128
Function
tokenize_function
(examples)
utils/data_utils.py:34
Function
train_client
(model, dataloader, optimizer, scheduler, args)
train_eval.py:37
Function
train_tokenize_function
(examples, tokenizer, query, response)
utils/data_utils.py:209
Method
unpack
(packed)
utils/offload_utils_for_quant/graph_hook.py:19
Function
update_decoder_weights
(target_module, new_weight)
utils/initialization_utils.py:57
← previous
101–134 of 134, ranked by callers