MCPcopy Create free account

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

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

⨍ Functions 183◇ Types & classes 38
↓ 1 callersMethodshould_save
(self)
SciScore_trainer/accelerators/base_accelerator.py:289
↓ 1 callersMethodshould_skip
(self, epoch, step)
SciScore_trainer/accelerators/base_accelerator.py:177
↓ 1 callersMethodtokenize
(self, example)
SciScore_trainer/datasetss/clip_hf_dataset.py:86
↓ 1 callersMethodtrain_step
(self, model, criterion, batch)
SciScore_trainer/tasks/clip_task.py:42
↓ 1 callersFunctionunpack_latents
(latents, height, width)
ft_flux/utils/diffusers_patch/flow_matching_with_logprob.py:12
↓ 1 callersMethodunwrap_and_save
(self, model)
SciScore_trainer/accelerators/base_accelerator.py:261
↓ 1 callersMethodupdate_epoch
(self)
SciScore_trainer/accelerators/base_accelerator.py:245
↓ 1 callersMethodupdate_global_step
(self, loss)
SciScore_trainer/accelerators/base_accelerator.py:214
↓ 1 callersMethodupdate_step
(self, loss, lr)
SciScore_trainer/accelerators/base_accelerator.py:227
↓ 1 callersMethodvalid_step
(self, model, criterion, batch)
SciScore_trainer/tasks/clip_task.py:58
↓ 1 callersFunctionverify_or_write_config
(cfg: TrainerConfig)
SciScore_trainer/train.py:49
FunctionImageReward
(device: str = "cuda")
ft_flux/utils/rewards.py:12
FunctionSciScore
(device: str = "cuda")
ft_flux/utils/rewards.py:24
Method__call__
(self, images, prompts)
ft_flux/utils/ImageReward.py:48
Method__call__
(self, images)
ft_flux/utils/aesthetic_scorer.py:45
Method__call__
(self, images, prompts, objects)
ft_flux/utils/SciScore.py:46
Method__getitem__
(self, idx)
SciScore_trainer/datasetss/clip_hf_dataset.py:107
Method__init__
(self, cfg: CLIPCriterionConfig)
SciScore_trainer/criterions/clip_criterion.py:19
Method__init__
(self, cfg: BaseAcceleratorConfig)
SciScore_trainer/accelerators/base_accelerator.py:76
Method__init__
(self, cfg: DebugAcceleratorConfig)
SciScore_trainer/accelerators/debug_accelerator.py:12
Method__init__
(self, cfg: DeepSpeedAcceleratorConfig)
SciScore_trainer/accelerators/deepspeed_accelerator.py:66
Method__init__
(self, model, lr=0.001, weight_decay=0, **kwargs)
SciScore_trainer/optimizers/dummy_optimizer.py:12
Method__init__
(self, cfg: CLIPTaskConfig, accelerator: BaseAccelerator)
SciScore_trainer/tasks/clip_task.py:36
Method__init__
(self, cfg: BaseTaskConfig, accelerator)
SciScore_trainer/tasks/base_task.py:23
Method__init__
(self, cfg: ClipModelConfig)
SciScore_trainer/models/clip_model.py:16
Method__init__
(self, cfg: CLIPHFDatasetConfig, split: str = "train")
SciScore_trainer/datasetss/clip_hf_dataset.py:54
Method__init__
(self, dtype, device)
ft_flux/utils/ImageReward.py:29
Method__init__
(self, dtype, device)
ft_flux/utils/aesthetic_scorer.py:31
Method__init__
(self, dtype, device)
ft_flux/utils/SciScore.py:33
Method__init__
(self, device, dtype)
eval/eval_t2i_with_SciScore.py:13
Method__len__
(self)
SciScore_trainer/datasetss/clip_hf_dataset.py:188
Function_fn
(images, prompts, metadata)
ft_flux/utils/rewards.py:7
Functionaesthetic
()
ft_flux/utils/prompts.py:49
Functionaesthetic_score
(device: str = "cuda")
ft_flux/utils/rewards.py:37
Functioncollate_fn
(examples)
ft_flux/sft_flux.py:270
Methodcollate_fn
(self, batch)
SciScore_trainer/datasetss/clip_hf_dataset.py:156
Methoddevice
(self)
SciScore_trainer/accelerators/base_accelerator.py:359
Methodevaluate
(self, model, criterion, dataloader)
SciScore_trainer/tasks/base_task.py:33
Methodforward
(self, model, batch)
SciScore_trainer/criterions/clip_criterion.py:87
Methodforward
(self, text_inputs=None, image_inputs=None)
SciScore_trainer/models/clip_model.py:26
Methodforward
(self, embed)
ft_flux/utils/ImageReward.py:24
Methodforward
(self, embed)
ft_flux/utils/aesthetic_scorer.py:26
Methodforward
(self, images, prompts)
eval/eval_t2i_with_SciScore.py:21
Functionget_config
(name)
ft_flux/config/custom.py:152
Functionget_config
()
ft_flux/config/base.py:4
Methodget_image_loss
(pos_logits, neg_logits)
SciScore_trainer/criterions/clip_criterion.py:71
Methodget_input_ids
(text)
SciScore_trainer/datasetss/clip_hf_dataset.py:87
Methodget_text_loss
(logits, align_pos=True)
SciScore_trainer/criterions/clip_criterion.py:60
Methodgradient_state
(self)
SciScore_trainer/accelerators/base_accelerator.py:331
Functionimagenet_all
()
ft_flux/utils/prompts.py:31
Functionimagenet_animals
()
ft_flux/utils/prompts.py:35
Functionimagenet_dogs
()
ft_flux/utils/prompts.py:39
Functioninstantiate_dummy_lr_scheduler
(cfg: ConstantWithWarmupLRSchedulerConfig, optimizer)
SciScore_trainer/lr_schedulers/constant_with_warmup.py:15
Functioninstantiate_dummy_lr_scheduler
(cfg: DummyLRSchedulerConfig, optimizer)
SciScore_trainer/lr_schedulers/dummy_lr_scheduler.py:23
Methodis_main_process
(self)
SciScore_trainer/accelerators/base_accelerator.py:142
Functionlight_reward
(device: str = "cuda")
ft_flux/utils/rewards.py:6
Methodload_best_checkpoint
(self)
SciScore_trainer/accelerators/base_accelerator.py:337
Functionload_model_hook
(models, input_dir)
ft_flux/oft_flux.py:165
Functionload_model_hook
(models, input_dir)
ft_flux/sft_flux.py:161
Methodload_state_if_needed
(self)
SciScore_trainer/accelerators/base_accelerator.py:124
Methodlogit_scale
(self)
SciScore_trainer/models/clip_model.py:36
Functionmain
(_)
ft_flux/oft_flux.py:40
Functionmain
(_)
ft_flux/sft_flux.py:40
Methodnum_processes
(self)
SciScore_trainer/accelerators/base_accelerator.py:146
Functionoft
()
ft_flux/config/custom.py:48
Functionoft_aes_score
()
ft_flux/config/custom.py:101
Functionoft_image_reward
()
ft_flux/config/custom.py:126
Functionoft_white
()
ft_flux/config/custom.py:75
Methodprepare
(self, *args, device_placement=None)
SciScore_trainer/accelerators/deepspeed_accelerator.py:95
Functionpreprocess_train
(examples)
ft_flux/sft_flux.py:251
Functionsave_model_hook
(models, weights, output_dir)
ft_flux/oft_flux.py:145
Functionsave_model_hook
(models, weights, output_dir)
ft_flux/sft_flux.py:141
Functionsci
()
ft_flux/utils/prompts.py:52
Functionsci_eval
()
ft_flux/utils/prompts.py:59
Functionscore_one_record
Score a single generated image via the LMM API.
eval/eval_t2i_with_lmm.py:103
Functionsimple_animals
()
ft_flux/utils/prompts.py:43
Functionsimple_animals_test
()
ft_flux/utils/prompts.py:46
Functionstandard_normal_logprob
(x)
ft_flux/utils/diffusers_patch/flux_pipeline_with_logprob.py:91
Methodsync_gradients
(self)
SciScore_trainer/accelerators/base_accelerator.py:208
Methodtrain_step
(self, model, criterion, batch)
SciScore_trainer/tasks/base_task.py:27
Methodtraining_stage
(self)
SciScore_trainer/accelerators/base_accelerator.py:293
Methodupdate_step_loss
(self, loss)
SciScore_trainer/accelerators/base_accelerator.py:211
Methodvalid_step
(self, model, criterion, batch)
SciScore_trainer/tasks/base_task.py:30
← previous101–183 of 183, ranked by callers