MCPcopy Create free account

hub / github.com/Kwai-Klear/AR-GRPO / functions

Functions650 in github.com/Kwai-Klear/AR-GRPO

Methodlist_loras
(self)
autoregressive/serve/model_runner.py:976
Methodlist_loras
(self)
autoregressive/serve/gpu_executor.py:178
Methodlist_loras
(self)
autoregressive/serve/llm_engine.py:667
Methodlist_loras
(self)
autoregressive/serve/worker.py:263
Methodload_model
(self, args)
autoregressive/serve/model_runner.py:161
Functionlogits_to_probs
(logits, temperature: float = 1.0, top_p: float=1.0, top_k: int = None, **kwargs)
autoregressive/models/generate.py:69
Methodmax_model_len
(self)
autoregressive/serve/worker.py:267
Methodmove_to_device
(self, device)
img_gen_grpo_rewards.py:40
Functionnon_saturating_d_loss
(logits_real, logits_fake)
tokenizer_lg/tokenizer_image/vq_loss.py:28
Functionnon_saturating_gen_loss
(logit_fake)
tokenizer_lg/tokenizer_image/vq_loss.py:39
Functionnormalize_01_into_pm1
(x)
lazy_dataset.py:11
Functionpickscore
(images, prompts, model=None)
reward_utils/pickscore.py:66
Functionpost_fork
(server, worker)
benchmark/reward-server/gunicorn.conf.py:18
Functionpre_fork
(server, worker)
benchmark/reward-server/gunicorn.conf.py:12
Functionprecompute_freqs_cis
(seq_len: int, n_elem: int, base: int = 10000, cls_token_num=120)
autoregressive/serve/gpt_model.py:334
Functionprecompute_freqs_cis
(seq_len: int, n_elem: int, base: int = 10000, cls_token_num=120)
autoregressive/models/gpt.py:410
Methodprediction_step
(self, model, inputs, prediction_loss_only, ignore_keys: Optional[list[str]] = None)
modified_grpo_trainer.py:1460
Methodprepare_inputs_for_generation
( self )
autoregressive/modeling.py:205
Functionquestion_image_score_prompt_key_pipe
(img, prompt_key, pipe=None, model_name=default_model_name, gen_kwargs={"max_new_tokens":256}, messages=defalu
reward_utils/qwen_vl_reward.py:139
Functionquestion_image_score_with_lbl_pipe
(img,lbl, pipe=None, model_name=default_model_name, gen_kwargs={"max_new_tokens":256}, messages=defalut_messag
reward_utils/qwen_vl_reward.py:93
Functionquestion_image_score_with_lbl_vllm
(img,lbl, client=None, model_name=default_model_name, gen_kwargs={"max_new_tokens":256}, messages=defalut_mess
reward_utils/qwen_vl_reward.py:62
Methodread_batch
(self, batch_size: int)
fid_evaluation.py:478
Methodread_batch
(self, batch_size: int)
fid_evaluation.py:508
Methodremaining
(self)
fid_evaluation.py:493
Methodremaining
(self)
fid_evaluation.py:516
Methodremove_lora
(self, lora_id: int)
autoregressive/serve/model_runner.py:971
Methodremove_lora
(self, lora_id: int)
autoregressive/serve/gpu_executor.py:174
Methodremove_lora
(self, lora_id: int)
autoregressive/serve/llm_engine.py:664
Methodremove_lora
(self, lora_id: int)
autoregressive/serve/worker.py:260
Functionreward_quantize
(value,min_v,max_v,min_q=0,max_q=5,num_bins=10)
img_gen_grpo_rewards.py:25
Functionsave_gif_grid
(video, file_name, nrow=None, fps=5)
utils/video.py:89
Functionsave_npz
Builds a single .npz file from a folder of .png samples. Refer to DiT.
test_t2i.py:91
Functionsave_npz
Builds a single .npz file from a folder of .png samples. Refer to DiT.
test_t2i_rewards.py:118
Methodsave_pretrained
(self,out_dir)
autoregressive/modeling.py:79
Functionsave_video_grid
(video, fname, nrow=None, fps=5)
utils/video.py:67
Functionscore
(img_or_path: Union[list, str, Image.Image], prompt: str, model=None, preprocessor=None, tokenizer=None)
reward_utils/hpsv2.py:48
Methodset_tokenizer
( self, tokenizer: Union[PreTrainedTokenizer, PreTrainedTokenizerFast], )
autoregressive/serve/llm.py:132
Functionshift_dim
(x, src_dim=-1, dest_dim=-1, make_contiguous=True)
utils/video.py:8
Functiontensor_slice
(x, begin, size)
utils/video.py:57
Functiontimed
(fn)
benchmark/reward-server/reward_server/gen_eval.py:28
Functiontimed
(fn)
benchmark/geneval/evaluation/evaluate_images.py:49
Methodtotal_params
(self)
autoregressive/modeling.py:217
Methodtrainable_params
(self)
autoregressive/modeling.py:221
Functionvanilla_d_loss
(logits_real, logits_fake)
tokenizer_lg/tokenizer_image/vq_loss.py:21
Functionview_range
(x, i, j, shape)
utils/video.py:38
Methodvocab_size
(self)
autoregressive/serve/model_runner.py:1094
Methodvocab_size
(self)
autoregressive/serve/worker.py:271
Methodwarmup
(self)
fid_evaluation.py:245
Functionwrapper
(*args, **kwargs)
benchmark/reward-server/reward_server/gen_eval.py:29
Functionwrapper
(*args, **kwargs)
benchmark/geneval/evaluation/evaluate_images.py:50
← previous601–650 of 650, ranked by callers