Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/EmbodiedGPT/EmbodiedGPT_Pytorch
/ functions
Functions
424 in github.com/EmbodiedGPT/EmbodiedGPT_Pytorch
⨍
Functions
424
◇
Types & classes
107
↓ 1 callers
Method
get_output_embeddings
(self)
robohusky/model/modeling_husky.py:1296
↓ 1 callers
Method
get_output_embeddings
(self)
robohusky/model/modeling_husky_embody2.py:1438
↓ 1 callers
Method
get_output_embeddings
(self)
robohusky/model/modeling_husky_embody2_ori.py:1297
↓ 1 callers
Method
get_video_embedding
(self, video_file)
demo/test.py:310
↓ 1 callers
Method
get_video_embedding
(self, video_file)
demo/script.py:357
↓ 1 callers
Method
get_video_embedding
(self, video_file)
demo/inference.py:309
↓ 1 callers
Function
iou
(box1, box2)
demo/script.py:71
↓ 1 callers
Function
is_image
(image_file)
robohusky/base_dataset_uni.py:58
↓ 1 callers
Function
is_numpy
(image_file)
robohusky/base_dataset_uni.py:70
↓ 1 callers
Function
is_video
(image_file)
robohusky/base_dataset_uni.py:64
↓ 1 callers
Function
load_image
(image, input_size=224)
demo/script.py:195
↓ 1 callers
Function
load_model
( model_path, device, num_gpus, max_gpu_memory=None, load_8bit=False, lora_weights=None )
demo/test.py:72
↓ 1 callers
Function
load_model
( model_path, device, num_gpus, max_gpu_memory=None, load_8bit=False, lora_weights=None )
demo/script.py:132
↓ 1 callers
Function
load_model
( model_path, device, num_gpus, max_gpu_memory=None, load_8bit=False, lora_weights=None )
demo/inference.py:72
↓ 1 callers
Function
load_video
(video_path, num_segments=8)
demo/script.py:208
↓ 1 callers
Function
main
()
robohusky/train/train.py:329
↓ 1 callers
Function
main
()
robohusky/train/train_uni.py:328
↓ 1 callers
Method
prune_heads
(self, heads)
robohusky/model/modeling_husky.py:773
↓ 1 callers
Method
prune_heads
(self, heads)
robohusky/model/modeling_husky_embody2.py:918
↓ 1 callers
Method
prune_heads
(self, heads)
robohusky/model/modeling_husky_embody2_ori.py:777
↓ 1 callers
Method
reduce_boxes
(self,scores,boxes,names,iou_threshold = 0.5)
demo/script.py:291
↓ 1 callers
Method
refine_bbox_dict
(self,boxes, names)
demo/script.py:313
↓ 1 callers
Method
save_attention_map
(self, attention_map)
robohusky/model/modeling_husky.py:656
↓ 1 callers
Method
save_attention_map
(self, attention_map)
robohusky/model/modeling_husky_embody2.py:732
↓ 1 callers
Method
save_attention_map
(self, attention_map)
robohusky/model/modeling_husky_embody2_ori.py:660
↓ 1 callers
Method
set_input_embeddings
(self, value)
robohusky/model/modeling_husky.py:1290
↓ 1 callers
Method
set_input_embeddings
(self, value)
robohusky/model/modeling_husky_embody2.py:1432
↓ 1 callers
Method
set_input_embeddings
(self, value)
robohusky/model/modeling_husky_embody2_ori.py:1291
↓ 1 callers
Method
set_output_embeddings
(self, new_embeddings)
robohusky/model/modeling_husky.py:1293
↓ 1 callers
Method
set_output_embeddings
(self, new_embeddings)
robohusky/model/modeling_husky_embody2.py:1435
↓ 1 callers
Method
set_output_embeddings
(self, new_embeddings)
robohusky/model/modeling_husky_embody2_ori.py:1294
↓ 1 callers
Function
test
()
robohusky/llama2_flash_attn_monkey_patch.py:139
↓ 1 callers
Function
test
()
robohusky/train/llama_flash_attn_monkey_patch.py:139
Method
__call__
(self, img_group)
robohusky/video_transformers.py:16
Method
__call__
(self, img_group)
robohusky/video_transformers.py:43
Method
__call__
(self, img_group)
robohusky/video_transformers.py:67
Method
__call__
(self, img_group, is_flow=False)
robohusky/video_transformers.py:77
Method
__call__
(self, tensor)
robohusky/video_transformers.py:94
Method
__call__
(self, img_group)
robohusky/video_transformers.py:116
Method
__call__
(self, img_group)
robohusky/video_transformers.py:130
Method
__call__
(self, img_group)
robohusky/video_transformers.py:170
Method
__call__
(self, img_group)
robohusky/video_transformers.py:217
Method
__call__
(self, img_group)
robohusky/video_transformers.py:305
Method
__call__
(self, images)
robohusky/video_transformers.py:346
Method
__call__
(self, img_group)
robohusky/video_transformers.py:360
Method
__call__
(self, pic)
robohusky/video_transformers.py:380
Method
__call__
(self, data)
robohusky/video_transformers.py:397
Method
__call__
This method uses [`BlipImageProcessor.__call__`] method to prepare image(s) for the model, and [`BertTokenizerFast.__call__`] to pr
robohusky/model/processing_husky.py:54
Method
__call__
(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs)
demo/test.py:177
Method
__call__
(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs)
demo/script.py:240
Method
__call__
(self, input_ids: torch.LongTensor, scores: torch.FloatTensor, **kwargs)
demo/inference.py:177
Method
__getattr__
(self, attr)
robohusky/utils.py:83
Method
__getitem__
(self, i)
robohusky/base_dataset.py:156
Method
__getitem__
(self, i)
robohusky/base_dataset.py:206
Method
__getitem__
(self, i)
robohusky/base_dataset_uni.py:267
Method
__init__
(self, logger, log_level=logging.INFO)
robohusky/utils.py:77
Method
__init__
(self, dataset, processor, input_size=224)
robohusky/base_dataset.py:184
Method
__init__
(self, size)
robohusky/video_transformers.py:10
Method
__init__
(self, size, groups=1)
robohusky/video_transformers.py:36
Method
__init__
(self, size)
robohusky/video_transformers.py:64
Method
__init__
(self, is_flow=False)
robohusky/video_transformers.py:74
Method
__init__
(self, mean, std)
robohusky/video_transformers.py:90
Method
__init__
(self, size, interpolation=Image.BILINEAR)
robohusky/video_transformers.py:113
Method
__init__
(self, crop_size, scale_size=None, flip=True)
robohusky/video_transformers.py:120
Method
__init__
(self, crop_size, scale_size=None, flip=True)
robohusky/video_transformers.py:160
Method
__init__
(self, input_size, scales=None, max_distort=1, fix_crop=True, more_fix_crop=True)
robohusky/video_transformers.py:207
Method
__init__
(self, size, interpolation=Image.BILINEAR)
robohusky/video_transformers.py:301
Method
__init__
(self, model_type)
robohusky/video_transformers.py:343
Method
__init__
(self, roll=False)
robohusky/video_transformers.py:357
Method
__init__
(self, div=True)
robohusky/video_transformers.py:377
Method
__init__
(self, weight=None, bias=None, device=None)
robohusky/compression.py:35
Method
__init__
( self, hidden_size=1408, intermediate_size=6144, projection_d
robohusky/configuration_husky.py:73
Method
__init__
( self, vocab_size=30522, hidden_size=768, num_hidden_layers=1
robohusky/configuration_husky.py:178
Method
__init__
( self, datasets: List[Dataset], weights: Sequence[float] = None,
robohusky/base_dataset_uni.py:345
Method
__init__
( self, weights: Sequence[float], num_samples: int, batch_size
robohusky/base_dataset_uni.py:395
Method
__init__
(self, config: HuskyVisionConfig)
robohusky/model/modeling_husky.py:89
Method
__init__
(self, config: HuskyVisionConfig)
robohusky/model/modeling_husky.py:121
Method
__init__
(self, config)
robohusky/model/modeling_husky.py:159
Method
__init__
(self, config: HuskyConfig)
robohusky/model/modeling_husky.py:257
Method
__init__
(self, config: HuskyConfig)
robohusky/model/modeling_husky.py:468
Method
__init__
(self, config: HuskyVisionConfig)
robohusky/model/modeling_husky.py:554
Method
__init__
(self, config, is_cross_attention=False)
robohusky/model/modeling_husky.py:622
Method
__init__
(self, config)
robohusky/model/modeling_husky.py:754
Method
__init__
(self, config, is_cross_attention=False)
robohusky/model/modeling_husky.py:767
Method
__init__
(self, config)
robohusky/model/modeling_husky.py:816
Method
__init__
(self, config)
robohusky/model/modeling_husky.py:831
Method
__init__
(self, config, layer_idx)
robohusky/model/modeling_husky.py:844
Method
__init__
(self, config)
robohusky/model/modeling_husky.py:943
Method
__init__
(self, config: HuskyQFormerConfig)
robohusky/model/modeling_husky.py:1048
Method
__init__
(self, config)
robohusky/model/modeling_husky.py:1236
Method
__init__
(self, config: HuskyConfig)
robohusky/model/modeling_husky.py:1269
Method
__init__
(self, config: HuskyConfig)
robohusky/model/modeling_husky.py:1509
Method
__init__
( self, hidden_size=1408, intermediate_size=6144, projecti
robohusky/model/configuration_husky_ori.py:74
Method
__init__
( self, vocab_size=30522, hidden_size=768, num_hidden_laye
robohusky/model/configuration_husky_ori.py:179
Method
__init__
(self, config: HuskyVisionConfig)
robohusky/model/modeling_husky_embody2.py:99
Method
__init__
(self, config: HuskyVisionConfig)
robohusky/model/modeling_husky_embody2.py:131
Method
__init__
(self, config)
robohusky/model/modeling_husky_embody2.py:169
Method
__init__
(self, config: HuskyConfig)
robohusky/model/modeling_husky_embody2.py:320
Method
__init__
(self, config: HuskyConfig)
robohusky/model/modeling_husky_embody2.py:540
Method
__init__
(self, config: HuskyVisionConfig)
robohusky/model/modeling_husky_embody2.py:626
← previous
next →
101–200 of 424, ranked by callers