Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/datawhalechina/all-in-rag
/ functions
Functions
495 in github.com/datawhalechina/all-in-rag
⨍
Functions
495
◇
Types & classes
94
Method
encode_text
ColBERT文本编码
code/C4/07_rerank_and_refine.py:33
Method
export_metadata
导出元数据到JSON文件 Args: output_path: 输出文件路径
code/C8/rag_modules/data_preparation.py:290
Method
extra_repr
(self)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:43
Method
filter_documents_by_category
按分类过滤文档 Args: category: 菜品分类 Returns: 过滤后的文档列表
code/C8/rag_modules/data_preparation.py:236
Method
filter_documents_by_difficulty
按难度过滤文档 Args: difficulty: 难度等级 Returns: 过滤后的文档列表
code/C8/rag_modules/data_preparation.py:248
Method
format_for_display
格式化用于显示的用量字符串 Args: amount: 用量 unit: 单位 Returns: 格式化后的字符串
code/C9/agent(代码系ai生成)/amount_normalizer.py:181
Method
forward
(self, mm_it_query=None, image_candidate=None, text_candidate=None, text_query=None, mm_it_candidate=None, tas
code/C3/visual_bge/visual_bge/modeling.py:320
Method
forward
(self, image_features, text_features, logit_scale=1.)
code/C3/visual_bge/visual_bge/eva_clip/loss.py:95
Method
forward
(self, t, start_index = 0)
code/C3/visual_bge/visual_bge/eva_clip/rope.py:70
Method
forward
(self, t, patch_indices_keep=None)
code/C3/visual_bge/visual_bge/eva_clip/rope.py:121
Method
forward
(ctx, tensor, rank, world_size)
code/C3/visual_bge/visual_bge/eva_clip/utils.py:311
Method
forward
(self, x: torch.Tensor)
code/C3/visual_bge/visual_bge/eva_clip/modified_resnet.py:42
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/modified_resnet.py:68
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/modified_resnet.py:173
Method
forward
(self, img)
code/C3/visual_bge/visual_bge/eva_clip/transform.py:24
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/timm_model.py:119
Method
forward
(self, x:BaseModelOutput, attention_mask:TensorType)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:46
Method
forward
(self, x:BaseModelOutput, attention_mask:TensorType)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:53
Method
forward
(self, x:BaseModelOutput, attention_mask:TensorType)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:65
Method
forward
(self, x:TensorType)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:213
Method
forward
(self, x: torch.Tensor)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:41
Method
forward
(self, x: torch.Tensor)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:55
Method
forward
(self, x: torch.Tensor)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:62
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:72
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:87
Method
forward
(self, x, attn_mask: Optional[torch.Tensor] = None)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:195
Method
forward
(self, query: torch.Tensor, key: torch.Tensor, value: torch.Tensor, attn_mask: Optional[torch.Tensor] = None)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:286
Method
forward
(self, q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, attn_mask: Optional[torch.Tensor] = None)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:384
Method
forward
(self, q: torch.Tensor, k: torch.Tensor = None, v: torch.Tensor = None, attn_mask: Optional[torch.Tensor] = No
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:432
Method
forward
(self, x: torch.Tensor, attn_mask: Optional[torch.Tensor] = None)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:480
Method
forward
(self, x: torch.Tensor, attn_mask: Optional[torch.Tensor] = None)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:511
Method
forward
(self, x: torch.Tensor, return_all_features: bool=False)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:611
Method
forward
(self, text, return_all_features: bool=False)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:722
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:40
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:70
Method
forward
(self, x)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:97
Method
forward
(self, x, rel_pos_bias=None, attn_mask=None)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:173
Method
forward
(self, x, rel_pos_bias=None, attn_mask=None)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:287
Method
forward
(self, x, **kwargs)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:320
Method
forward
(self)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:358
Method
forward
(self, x, return_all_features=True)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:527
Method
forward
(self, image, text)
code/C3/visual_bge/visual_bge/eva_clip/model.py:264
Method
forward
(self, image, text)
code/C3/visual_bge/visual_bge/eva_clip/model.py:316
Method
forward_mlm
(self, input_ids, image_embeds, mlm_probability=0.25)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:177
Method
from_dict
从字典创建配置对象
code/C8/config.py:32
Method
from_dict
从字典创建配置对象
code/C9/config.py:51
Method
get_cast_dtype
(self)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:429
Method
get_cast_dtype
(self)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:454
Method
get_classifier
(self)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:482
Method
get_comparable_value
获取可比较的数值(用于排序、筛选等) Args: amount: 用量 unit: 单位 Returns: 可比较的数值,None表示
code/C9/agent(代码系ai生成)/amount_normalizer.py:160
Method
get_image_paths
获取所有图像路径
code/C3/work_multimodal_dragon_search.py:45
Method
get_ingredients_list
获取指定菜品的食材信息 Args: dish_name: 菜品名称 Returns: 食材信息
code/C8/main.py:289
Method
get_metadata_by_path
根据路径获取元数据
code/C3/work_multimodal_dragon_search.py:49
Method
get_num_layers
(self)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:242
Method
get_num_layers
(self)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:600
Method
get_num_layers
(self)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:711
Method
get_num_layers
(self)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:466
Method
get_statistics
获取数据统计信息 Returns: 统计信息字典
code/C9/rag_modules/graph_data_preparation.py:413
Method
get_table_info
获取数据库表信息
code/C4/text2sql/text2sql_agent.py:164
Function
get_tokenizer
(model_name)
code/C3/visual_bge/visual_bge/eva_clip/factory.py:73
Method
init_parameters
(self)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:247
Method
initialize
初始化检索系统
code/C9/rag_modules/hybrid_retrieval.py:68
Function
is_logging
(args)
code/C3/visual_bge/visual_bge/eva_clip/utils.py:292
Function
is_master
(args, local=False)
code/C3/visual_bge/visual_bge/eva_clip/utils.py:299
Function
list_pretrained
returns list of pretrained models Returns a tuple (model_name, pretrain_tag) by default or 'name:tag' if as_str == True
code/C3/visual_bge/visual_bge/eva_clip/pretrained.py:196
Method
lock
(self, unlocked_groups=0, freeze_bn_stats=False)
code/C3/visual_bge/visual_bge/eva_clip/modified_resnet.py:154
Method
lock
(self, unlocked_layers:int=0, freeze_layer_norm:bool=True)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:220
Method
lock
(self, unlocked_groups=0, freeze_bn_stats=False)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:567
Method
lock
(self, unlocked_groups=0, freeze_bn_stats=False)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:469
Method
lock_image_tower
(self, unlocked_groups=0, freeze_bn_stats=False)
code/C3/visual_bge/visual_bge/eva_clip/model.py:233
Method
lock_image_tower
(self, unlocked_groups=0, freeze_bn_stats=False)
code/C3/visual_bge/visual_bge/eva_clip/model.py:291
Method
lock_text_tower
(self, unlocked_layers:int=0, freeze_layer_norm:bool=True)
code/C3/visual_bge/visual_bge/eva_clip/model.py:295
Method
no_weight_decay
(self)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:608
Method
no_weight_decay
(self)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:707
Method
no_weight_decay
(self)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:479
Method
no_weight_decay
(self)
code/C3/visual_bge/visual_bge/eva_clip/model.py:243
Method
no_weight_decay
(self)
code/C3/visual_bge/visual_bge/eva_clip/model.py:305
Function
parse
(x)
code/C3/visual_bge/visual_bge/eva_clip/utils.py:278
Method
parse_amount_with_unit
解析包含单位的用量字符串 Args: amount_str: 如"300毫升"、"适量盐"等 Returns: Tuple[数量, 单位, 估算数值]
code/C9/agent(代码系ai生成)/amount_normalizer.py:123
Function
register_pooler
Decorator registering pooler class
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:37
Method
rescale
(param, layer_id)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:444
Function
resize_rel_pos_embed
(state_dict, model, interpolation: str = 'bicubic', seq_dim=1)
code/C3/visual_bge/visual_bge/eva_clip/utils.py:140
Function
route
(info)
code/C4/06_embedding_based_routing.py:45
Method
save
(self, output_dir: str)
code/C3/visual_bge/visual_bge/modeling.py:377
Method
search_by_category
按分类搜索菜品 Args: category: 菜品分类 query: 可选的额外查询条件 Returns: 菜品名称列表
code/C8/main.py:260
Method
set_grad_checkpointing
(self, enable=True)
code/C3/visual_bge/visual_bge/eva_clip/modified_resnet.py:162
Method
set_grad_checkpointing
(self, enable=True)
code/C3/visual_bge/visual_bge/eva_clip/timm_model.py:113
Method
set_grad_checkpointing
(self, enable=True)
code/C3/visual_bge/visual_bge/eva_clip/hf_model.py:239
Method
set_grad_checkpointing
(self, enable=True)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:604
Method
set_grad_checkpointing
(self, enable=True)
code/C3/visual_bge/visual_bge/eva_clip/transformer.py:703
Method
set_grad_checkpointing
(self, enable=True)
code/C3/visual_bge/visual_bge/eva_clip/eva_vit_model.py:475
Method
set_grad_checkpointing
(self, enable=True)
code/C3/visual_bge/visual_bge/eva_clip/model.py:299
Method
to_dict
转换为字典
code/C8/config.py:36
Function
tokenize
Returns the tokenized representation of given input string(s) Parameters ---------- texts : Union[str, List[str]] An input s
code/C3/visual_bge/visual_bge/eva_clip/tokenizer.py:156
Function
trace_model
(model, batch_size=256, device=torch.device('cpu'))
code/C3/visual_bge/visual_bge/eva_clip/model.py:439
← previous
401–495 of 495, ranked by callers