MCPcopy Create free account

hub / github.com/brycedrennan/imaginAIry / functions

Functions2,992 in github.com/brycedrennan/imaginAIry

Methoddevice
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/multi_diffusion.py:79
Methoddevice
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/restart.py:105
Methoddevice
(self)
imaginairy/vendored/refiners/foundationals/segment_anything/image_encoder.py:80
Methoddiffuse_target
(self, x: Tensor, step: int, target: SDXLDiffusionTarget)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/multi_diffusion.py:13
Methoddiffuse_target
(self, x: Tensor, step: int, target: DiffusionTarget)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/multi_diffusion.py:14
Methoddiffuse_target
(self, x: Tensor, step: int, target: InpaintingDiffusionTarget)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/multi_diffusion.py:30
Functiondiffusers_controlnet_sd15_to_refiners_translator
()
imaginairy/weight_management/translators.py:42
Functiondiffusers_t2i_adapter_sd15_to_refiners_translator
()
imaginairy/weight_management/translators.py:62
Functiondiffusers_t2i_adapter_sdxl_to_refiners_translator
()
imaginairy/weight_management/translators.py:66
Functiondiffusers_unet_sd15_to_refiners_translator
()
imaginairy/weight_management/translators.py:30
Functiondisable_pytorch_lighting_custom_logging
()
imaginairy/utils/log_utils.py:461
Functiondisabled_train
For disabling train/eval mode. Overwrite `model.train` with this function to make sure train/eval mode does not change anymore.
imaginairy/modules/encoders.py:48
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
imaginairy/modules/diffusion/ddpm.py:72
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
imaginairy/modules/midas/api.py:39
Functiondisabled_train
Overwrite model.train with this function to make sure train/eval mode does not change anymore.
imaginairy/utils/__init__.py:289
Functiondisplay_weights_structure
given a weights path, display the structure of the weights.
scripts/weights_debug.py:32
Methoddouble_distortions
(self)
imaginairy/utils/data_distorter.py:102
Methoddown
(self)
imaginairy/vendored/refiners/fluxion/adapters/lora.py:36
Functiondownload_file
Downloads a file if it does not exist, optionally checking its SHA-256 hash.
imaginairy/vendored/k_diffusion/utils.py:55
Functiondraw_handpose
(canvas, all_hand_peaks, show_number=False)
imaginairy/img_processors/openpose.py:126
Methoddtype
(self)
imaginairy/vendored/clip/model.py:413
Methoddtype
(self)
imaginairy/vendored/refiners/fluxion/layers/chain.py:329
Methoddtype
(self)
imaginairy/vendored/refiners/fluxion/layers/module.py:164
Methoddtype
(self)
imaginairy/vendored/refiners/fluxion/layers/basics.py:203
Methoddtype
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/multi_diffusion.py:83
Methoddtype
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/restart.py:109
Methoddtype
(self)
imaginairy/vendored/refiners/foundationals/segment_anything/image_encoder.py:84
Functionedit_cmd
Edit an image via AI. Provide paths or URLs to images and directions on how to alter them. Example: aimg edit --prompt "make the dog re
imaginairy/cli/edit.py:48
Functionedit_demo_cmd
Make some fun pre-set edits to input photos.
imaginairy/cli/edit_demo.py:31
Functionedit_redir
()
imaginairy/http_app/app.py:59
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/self_attention_guidance.py:73
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/reference_only_control.py:131
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/image_prompt.py:395
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/t2i_adapter.py:194
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_xl/self_attention_guidance.py:34
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/self_attention_guidance.py:34
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/stable_diffusion_1/controlnet.py:165
Methodeject
(self)
imaginairy/vendored/refiners/foundationals/clip/concepts.py:145
Functionema_update
Incorporates updated model parameters into an exponential moving averaged version of a model. It should be called after each optimizer step.
imaginairy/vendored/k_diffusion/utils.py:88
Methodembedding_extender
(self)
imaginairy/vendored/refiners/foundationals/clip/concepts.py:117
Methodencode
(self, text)
imaginairy/modules/encoders.py:94
Methodencode
(self, text)
imaginairy/modules/encoders.py:232
Methodencode
(self, text)
imaginairy/modules/encoders.py:255
Methodencode
(self, x, *args, **kwargs)
imaginairy/modules/autoencoder.py:595
Methodencode
(self, text)
imaginairy/modules/clip_embedders.py:50
Methodencode
(self, text)
imaginairy/modules/clip_embedders.py:86
Methodencode
(self, *args, **kwargs)
imaginairy/modules/sgm/autoencoder.py:97
Methodencode
( self, x: torch.Tensor, return_reg_log: bool = False )
imaginairy/modules/sgm/autoencoder.py:481
Methodencode
(self, x: Any, *args, **kwargs)
imaginairy/modules/sgm/autoencoder.py:570
Methodencode
(self, text)
imaginairy/modules/sgm/encoders/modules.py:307
Methodencode
(self, text)
imaginairy/modules/sgm/encoders/modules.py:350
Methodencode
(self, text)
imaginairy/modules/sgm/encoders/modules.py:416
Methodencode
(self, text)
imaginairy/modules/sgm/encoders/modules.py:519
Methodencode
(self, text)
imaginairy/modules/sgm/encoders/modules.py:592
Methodencode
(self, text)
imaginairy/modules/sgm/encoders/modules.py:751
Methodencode
(self, text)
imaginairy/modules/sgm/encoders/modules.py:775
Methodencode
(self, x)
imaginairy/modules/sgm/encoders/modules.py:840
Methodend_timestep
(self)
imaginairy/vendored/refiners/foundationals/latent_diffusion/restart.py:86
Methodensure_parent
(self)
imaginairy/vendored/refiners/fluxion/layers/module.py:112
Functioneval_mode
A context manager that places a model into evaluation mode and restores the previous mode on exit.
imaginairy/vendored/k_diffusion/utils.py:81
Functionexception_handler
(request: Request, exc: Exception)
imaginairy/http_app/app.py:72
Methodfeed_forward_chunk
(self, attention_output)
imaginairy/vendored/blip/nlvr_encoder.py:478
Methodfeed_forward_chunk
(self, attention_output)
imaginairy/vendored/blip/med.py:431
Functionfile_filter
(path: str)
scripts/modal_entrypoint.py:20
Methodfilename
(self)
imaginairy/weight_management/conversion.py:30
Functionfilename_base_for_orig_outputs
(request)
tests/conftest.py:95
Functionfilename_base_for_outputs
(request)
tests/conftest.py:89
Methodfilepath
(self)
imaginairy/weight_management/conversion.py:34
Functionfilesafe_text
(t)
imaginairy/utils/log_utils.py:361
Functionfind_project_root
Traverse up from a starting path to find the project root. The project root is identified by the presence of a '.git' directory inside it.
imaginairy/utils/gitignore.py:7
Methodfind_state_dict_key
(module: Module, /)
imaginairy/vendored/refiners/fluxion/layers/chain.py:193
Functionfn
(tau)
imaginairy/modules/sgm/diffusionmodules/sampling_utils.py:14
Functionfn
(tau)
imaginairy/vendored/k_diffusion/sampling.py:369
Methodforward
(self, x)
tests/test_utils/test_model_cache.py:15
Methodforward
(self, batch, key=None, disable_dropout=False)
imaginairy/modules/encoders.py:27
Methodforward
(self, text)
imaginairy/modules/encoders.py:78
Methodforward
(self, text)
imaginairy/modules/encoders.py:132
Methodforward
(self, text)
imaginairy/modules/encoders.py:205
Methodforward
(self, text)
imaginairy/modules/encoders.py:258
Methodforward
(self, model)
imaginairy/modules/ema.py:37
Methodforward
( self, x: Tensor, step: int, *, clip_text_embedding: Tensor,
imaginairy/modules/refiners_sd.py:567
Methodforward
(self, x)
imaginairy/modules/attention.py:36
Methodforward
(self, x)
imaginairy/modules/attention.py:56
Methodforward
(self, x)
imaginairy/modules/attention.py:83
Methodforward
(self, x, context=None, mask=None)
imaginairy/modules/attention.py:137
Methodforward
(self, x, context=None, mask=None)
imaginairy/modules/attention.py:279
Methodforward
(self, x, context=None)
imaginairy/modules/attention.py:353
Methodforward
(self, x, context=None)
imaginairy/modules/attention.py:428
Methodforward
( self, x, timesteps=None, context=None, control=None, only_mi
imaginairy/modules/cldm.py:24
Methodforward
(self, x, hint, timesteps, context, **kwargs)
imaginairy/modules/cldm.py:335
Methodforward
(self, input, sample_posterior=True)
imaginairy/modules/autoencoder.py:293
Methodforward
(self, x, *args, **kwargs)
imaginairy/modules/autoencoder.py:606
Methodforward
(self, text)
imaginairy/modules/clip_embedders.py:34
Methodforward
(self, text)
imaginairy/modules/clip_embedders.py:79
Methodforward
(self, x)
imaginairy/modules/clip_embedders.py:132
Methodforward
(self, x)
imaginairy/modules/diffusion/openaimodel.py:55
Methodforward
Apply the module to `x` given `emb` timestep embeddings.
imaginairy/modules/diffusion/openaimodel.py:72
Methodforward
(self, x, emb, context=None)
imaginairy/modules/diffusion/openaimodel.py:84
Methodforward
(self, x)
imaginairy/modules/diffusion/openaimodel.py:115
Methodforward
(self, x)
imaginairy/modules/diffusion/openaimodel.py:140
← previousnext →2,001–2,100 of 2,992, ranked by callers