MCPcopy Create free account

hub / github.com/Jialuo-Li/Science-T2I / functions

Functions183 in github.com/Jialuo-Li/Science-T2I

↓ 17 callersMethodlog
(self, data)
SciScore_trainer/accelerators/base_accelerator.py:189
↓ 6 callersFunctionfrom_file
(path, low=None, high=None)
ft_flux/utils/prompts.py:26
↓ 6 callersFunctioninstantiate_with_cfg
(cfg: DictConfig, **kwargs)
SciScore_trainer/configs/configs.py:72
↓ 6 callersMethodprepare
(self, *args, device_placement=None)
SciScore_trainer/accelerators/base_accelerator.py:112
↓ 6 callersMethodsimple_collate
(self, batch, column_name)
SciScore_trainer/datasetss/clip_hf_dataset.py:150
↓ 5 callersMethodgather
(self, data)
SciScore_trainer/accelerators/base_accelerator.py:204
↓ 5 callersMethodgather_features
(features)
SciScore_trainer/criterions/clip_criterion.py:41
↓ 5 callersMethodget_image_features
(self, *args, **kwargs)
SciScore_trainer/models/clip_model.py:23
↓ 5 callersMethodsave
(self, path)
SciScore_trainer/models/clip_model.py:39
↓ 4 callersFunctioncalculate_accuracy
(acc_dict)
eval/eval_lmm.py:76
↓ 4 callersFunctionencode_prompt
r""" Args: prompt (`str` or `List[str]`, *optional*): prompt to be encoded prompt_2 (`str` or `List[str]`, *optional*
ft_flux/utils/diffusers_patch/flux_encoder.py:106
↓ 4 callersMethodget_all_ckpts
(self)
SciScore_trainer/accelerators/base_accelerator.py:334
↓ 4 callersMethodget_text_features
(self, *args, **kwargs)
SciScore_trainer/models/clip_model.py:20
↓ 4 callersFunctionsft
()
ft_flux/config/custom.py:8
↓ 4 callersMethodwait_for_everyone
(self)
SciScore_trainer/accelerators/base_accelerator.py:242
↓ 3 callersMethodaccumulate
(self, model)
SciScore_trainer/accelerators/base_accelerator.py:201
↓ 3 callersMethodbackward
(self, loss)
SciScore_trainer/accelerators/base_accelerator.py:274
↓ 3 callersMethodclip_grad_norm_
(self, params)
SciScore_trainer/accelerators/base_accelerator.py:277
↓ 3 callersFunctionflowmatching_with_logprob
Predict the sample from the previous timestep by reversing the SDE. This function propagates the diffusion process from the learned model out
ft_flux/utils/diffusers_patch/flow_matching_with_logprob.py:41
↓ 3 callersFunctionflux_pipeline_with_logprob
r""" Function invoked when calling the pipeline for generation. Args: prompt (`str` or `List[str]`, *optional*): The prom
ft_flux/utils/diffusers_patch/flux_pipeline_with_logprob.py:95
↓ 2 callersFunctionevaluate
()
SciScore_trainer/train.py:94
↓ 2 callersMethodevaluate
(self, model, criterion, dataloader)
SciScore_trainer/tasks/clip_task.py:105
↓ 2 callersFunctionevaluate_dataset
(dataset, tokenizer, model, image_processor, reverse)
eval/eval_lmm.py:51
↓ 2 callersMethodget_features
(model, input_ids, pos_input_ids, neg_input_ids, pixels_pos_values, pixels_neg_values)
SciScore_trainer/criterions/clip_criterion.py:24
↓ 2 callersFunctionload_model
(model_pretrained_name, processor_pretrained_name, device)
eval/eval_vlm.py:59
↓ 2 callersMethodnormalize
(features)
SciScore_trainer/criterions/clip_criterion.py:30
↓ 2 callersMethodpixel_values_to_pil_images
(pixel_values)
SciScore_trainer/tasks/clip_task.py:73
↓ 2 callersMethodpost_init
(self)
SciScore_trainer/accelerators/base_accelerator.py:90
↓ 2 callersMethodprocess_images
(self, images)
SciScore_trainer/datasetss/clip_hf_dataset.py:103
↓ 2 callersMethodsave_checkpoint
(self)
SciScore_trainer/accelerators/base_accelerator.py:306
↓ 2 callersMethodsave_training_stage
(self, save_dir)
SciScore_trainer/accelerators/base_accelerator.py:303
↓ 2 callersMethodshould_end
(self)
SciScore_trainer/accelerators/base_accelerator.py:271
↓ 2 callersFunctionunwrap_model
(model)
ft_flux/oft_flux.py:139
↓ 2 callersFunctionunwrap_model
(model)
ft_flux/sft_flux.py:135
↓ 2 callersMethodupdate_metrics
(self, metrics)
SciScore_trainer/accelerators/base_accelerator.py:252
↓ 2 callersMethodupdate_progbar_step
(self)
SciScore_trainer/accelerators/base_accelerator.py:186
↓ 1 callersMethod__init__
(self)
ft_flux/utils/ImageReward.py:10
↓ 1 callersMethod__init__
(self)
ft_flux/utils/aesthetic_scorer.py:12
↓ 1 callersFunction_flatten_dict
(params: MutableMapping, delimiter: str = "/", parent_key: str = "")
SciScore_trainer/accelerators/utils.py:60
↓ 1 callersFunction_get_clip_prompt_embeds
( self, prompt: Union[str, List[str]], num_images_per_prompt: int = 1, device: Optional[torch.
ft_flux/utils/diffusers_patch/flux_encoder.py:63
↓ 1 callersFunction_get_t5_prompt_embeds
( self, prompt: Union[str, List[str]] = None, num_images_per_prompt: int = 1, max_sequence_len
ft_flux/utils/diffusers_patch/flux_encoder.py:17
↓ 1 callersFunction_load_lines
Load lines from a file. First tries to load from `path` directly, and if that doesn't exist, searches the `d3po_pytorch/assets` directory for
ft_flux/utils/prompts.py:12
↓ 1 callersFunction_locate
Locate an object by name or dotted path, importing as necessary. This is similar to the pydoc function `locate`, except that it checks for
SciScore_trainer/configs/configs.py:18
↓ 1 callersFunctionapply_box_mask
(tensor, box)
ft_flux/utils/diffusers_patch/flow_matching_with_logprob.py:22
↓ 1 callersFunctionbuild_parser
()
eval/eval_t2i_with_lmm.py:217
↓ 1 callersMethodcalc_loss
(self, text_features, text_pos_features, text_neg_features, image_pos_features, image_neg_features, logit_scal
SciScore_trainer/criterions/clip_criterion.py:44
↓ 1 callersFunctioncalculate_accuracy
(probs, dataset)
eval/eval_vlm.py:9
↓ 1 callersFunctioncalculate_category_accuracy
(probs, categories)
eval/eval_vlm.py:13
↓ 1 callersFunctioncalculate_law_accuracy
(probs, laws)
eval/eval_vlm.py:21
↓ 1 callersFunctioncalculate_shift
( image_seq_len, base_seq_len: int = 256, max_seq_len: int = 4096, base_shift: float = 0.5,
ft_flux/utils/diffusers_patch/flux_pipeline_with_logprob.py:20
↓ 1 callersFunctioncalculate_similarity
(images, prompt, model, processor, device)
eval/eval_vlm.py:33
↓ 1 callersMethodcleanup_checkpoints
(self)
SciScore_trainer/accelerators/base_accelerator.py:362
↓ 1 callersFunctioncompute_scores
Score all records via LMM API and aggregate by category/law.
eval/eval_t2i_with_lmm.py:166
↓ 1 callersFunctioncompute_scores
(pipe, sci_scorer, prompt, ip, num_samples=2)
eval/eval_t2i_with_SciScore.py:53
↓ 1 callersFunctiondebug
(port)
SciScore_trainer/accelerators/base_accelerator.py:27
↓ 1 callersMethoddebug
(self)
SciScore_trainer/accelerators/base_accelerator.py:104
↓ 1 callersMethodend_training
(self)
SciScore_trainer/accelerators/base_accelerator.py:257
↓ 1 callersFunctionevaluate_dataset
(dataset, model, processor, device)
eval/eval_vlm.py:66
↓ 1 callersFunctionevaluate_results
(dataset_name, processor_pretrained_name, model_pretrained_name)
eval/eval_vlm.py:82
↓ 1 callersMethodfeatures2probs
(model, text_features, image_pos_features, image_neg_features)
SciScore_trainer/tasks/clip_task.py:47
↓ 1 callersFunctionfind_enclosing_box_cxcywh
(box_tensor)
ft_flux/utils/SciScore.py:6
↓ 1 callersFunctionflatten
(list_of_lists)
SciScore_trainer/tasks/base_task.py:11
↓ 1 callersMethodgather_dict
(self, eval_dict)
SciScore_trainer/tasks/base_task.py:69
↓ 1 callersMethodgather_iterable
(it, num_processes)
SciScore_trainer/tasks/base_task.py:59
↓ 1 callersFunctiongenerate_images
Generate images for each test sample and save metadata.
eval/eval_t2i_with_lmm.py:117
↓ 1 callersMethodget_allocated_cuda_memory
(self)
SciScore_trainer/accelerators/base_accelerator.py:224
↓ 1 callersMethodget_ckpts_to_delete
(self)
SciScore_trainer/accelerators/base_accelerator.py:381
↓ 1 callersFunctionget_dataset_preprocessor
()
ft_flux/sft_flux.py:242
↓ 1 callersMethodget_latest_checkpoint
(self)
SciScore_trainer/accelerators/base_accelerator.py:115
↓ 1 callersFunctionget_nvidia_smi_gpu_memory_stats_str
()
SciScore_trainer/accelerators/utils.py:42
↓ 1 callersFunctionget_response
(data, tokenizer, model, image_processor, reverse)
eval/eval_lmm.py:15
↓ 1 callersFunctionget_sigmas
(timesteps, n_dim=4, dtype=torch.float32)
ft_flux/sft_flux.py:389
↓ 1 callersMethodinit_training
(self, cfg: DictConfig)
SciScore_trainer/accelerators/base_accelerator.py:164
↓ 1 callersFunctionload_dataloaders
(cfg: DictConfig)
SciScore_trainer/train.py:19
↓ 1 callersMethodload_hf_dataset
(self, split: str)
SciScore_trainer/datasetss/clip_hf_dataset.py:75
↓ 1 callersFunctionload_meta_records
(output_dir)
eval/eval_t2i_with_lmm.py:158
↓ 1 callersFunctionload_optimizer
(cfg: DictConfig, model: nn.Module)
SciScore_trainer/train.py:34
↓ 1 callersFunctionload_scheduler
(cfg: DictConfig, optimizer)
SciScore_trainer/train.py:39
↓ 1 callersFunctionload_task
(cfg: DictConfig, accelerator: BaseAccelerator)
SciScore_trainer/train.py:44
↓ 1 callersMethodlog_to_wandb
(self, eval_dict, table_name="test_predictions")
SciScore_trainer/tasks/base_task.py:36
↓ 1 callersFunctionmain
(cfg: TrainerConfig)
SciScore_trainer/train.py:62
↓ 1 callersFunctionmain
(dataset_name)
eval/eval_lmm.py:83
↓ 1 callersFunctionmain
(args)
eval/eval_t2i_with_lmm.py:193
↓ 1 callersFunctionmain
(dataset_name)
eval/eval_t2i_with_SciScore.py:61
↓ 1 callersFunctionnumpy_to_pil
(images)
SciScore_trainer/tasks/clip_task.py:29
↓ 1 callersFunctionnvidia_smi_gpu_memory_stats
Parse the nvidia-smi output and extract the memory used stats.
SciScore_trainer/accelerators/utils.py:12
↓ 1 callersFunctionparse_score_response
Extract overall/reality score from LMM JSON response.
eval/eval_t2i_with_lmm.py:32
↓ 1 callersMethodpre_training_log
(self, cfg: DictConfig)
SciScore_trainer/accelerators/base_accelerator.py:149
↓ 1 callersFunctionprint_config
(cfg: DictConfig)
SciScore_trainer/accelerators/utils.py:46
↓ 1 callersMethodrecalc_train_length_after_prepare
(self, num_batches)
SciScore_trainer/accelerators/base_accelerator.py:193
↓ 1 callersFunctionrequest_openai_chat
Send image + prompt to an OpenAI-compatible multimodal API.
eval/eval_t2i_with_lmm.py:70
↓ 1 callersFunctionretrieve_timesteps
Calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. Handles custom timesteps. Any kwargs
ft_flux/utils/diffusers_patch/flux_pipeline_with_logprob.py:33
↓ 1 callersMethodrun_inference
(self, model, criterion, dataloader)
SciScore_trainer/tasks/clip_task.py:79
↓ 1 callersFunctionsave_and_prepare_image
(image, tmpdir, filename)
ft_flux/oft_flux.py:375
↓ 1 callersFunctionsave_and_prepare_image
(image, tmpdir, filename)
ft_flux/sft_flux.py:354
↓ 1 callersMethodselect_combination
(self, num_captions, num_pos_imgs, num_neg_imgs, num_pos_captions, num_neg_captions)
SciScore_trainer/datasetss/clip_hf_dataset.py:66
↓ 1 callersMethodset_logging_level
(self)
SciScore_trainer/accelerators/base_accelerator.py:96
↓ 1 callersMethodset_mixed_precision
(self)
SciScore_trainer/accelerators/deepspeed_accelerator.py:84
↓ 1 callersMethodset_seed
(self)
SciScore_trainer/accelerators/base_accelerator.py:108
↓ 1 callersMethodshould_eval
(self)
SciScore_trainer/accelerators/base_accelerator.py:280
next →1–100 of 183, ranked by callers