CodeMuse is now Onedoc, a document generation platform

Spend less time
onboarding a repository

Search your repository in natural language,
browse automatically generated, up-to-date documentation.
Right now.

1Search

Or try one of...

CLIP model implementation

Convert numpy array to pil image

Image label viewer

2Results

CLIP

1.19

This code defines a class called "SD Model" that is used for image-text matching. It has various functions, including `get_model`, `get_input_names`, and `get_output_names`, which are used to retrieve information about the model and its inputs and outputs. Additionally, it has a function called `optimize` that optimizes an ONNX graph and exports it as an ONNX model if requested.

View on GitHub

/src/streamdiffusion/acceleration/tensorrt/models.py

CLIP#__init__()

1.02

/src/streamdiffusion/acceleration/tensorrt/models.py

UNet

0.37

/src/streamdiffusion/acceleration/tensorrt/models.py

VAE

0.35

/src/streamdiffusion/acceleration/tensorrt/models.py

CLIP#optimize()

0.35

/src/streamdiffusion/acceleration/tensorrt/models.py

Optimizer#infer_shapes()

0.33

/src/streamdiffusion/acceleration/tensorrt/models.py

BaseModel

0.33

/src/streamdiffusion/acceleration/tensorrt/models.py

create_models()

0.33

/src/streamdiffusion/acceleration/tensorrt/utilities.py

VAEEncoder

0.33

/src/streamdiffusion/acceleration/tensorrt/models.py

UNet#get_input_profile()

0.32

/src/streamdiffusion/acceleration/tensorrt/models.py

Optimizer

0.30

/src/streamdiffusion/acceleration/tensorrt/models.py

VAE#get_input_profile()

0.28

/src/streamdiffusion/acceleration/tensorrt/models.py

CLIP#get_input_profile()

0.28

/src/streamdiffusion/acceleration/tensorrt/models.py

BaseModel#optimize()

0.27

/src/streamdiffusion/acceleration/tensorrt/models.py

VAEEncoder#get_input_profile()

0.27

/src/streamdiffusion/acceleration/tensorrt/models.py

StreamDiffusionWrapper#_load_model()

0.25

/utils/wrapper.py

__init__

0.25

/src/streamdiffusion/acceleration/tensorrt/models.py

main()

0.23

/examples/img2img/single.py

StreamDiffusionWrapper

0.23

/utils/wrapper.py

main()

0.23

/examples/img2img/multi.py

StreamDiffusionWrapper#prepare()

0.23

/utils/wrapper.py

__init__

0.23

/src/streamdiffusion/acceleration/tensorrt/utilities.py

main()

0.23

/examples/screen/main.py

SimilarImageFilter

0.23

/src/streamdiffusion/image_filter.py

StreamDiffusionWrapper#create_prefix()

0.23

/utils/wrapper.py

build_engine()

0.23

/src/streamdiffusion/acceleration/tensorrt/utilities.py

__init__

0.22

/examples/img2img/multi.py

SimilarImageFilter#__call__()

0.22

/src/streamdiffusion/image_filter.py

__init__

0.22

/examples/img2img/single.py

StreamDiffusion#__init__()

0.22

/src/streamdiffusion/pipeline.py

StreamDiffusionWrapper#img2img()

0.21

/utils/wrapper.py

StreamDiffusion

0.21

/src/streamdiffusion/pipeline.py

UNet#get_sample_input()

0.21

/src/streamdiffusion/acceleration/tensorrt/models.py

image_generation_process()

0.21

/examples/screen/main.py

Optimizer#info()

0.21

/src/streamdiffusion/acceleration/tensorrt/models.py

main()

0.20

/examples/vid2vid/main.py

__init__

0.20

/src/streamdiffusion/pipeline.py

__init__

0.20

/examples/vid2vid/main.py

PredictInputModel

0.20

/demo/realtime-txt2img/server/main.py

StreamDiffusionWrapper#__init__()

0.20

/utils/wrapper.py

__init__

0.20

/src/streamdiffusion/acceleration/tensorrt/engine.py

__init__

0.20

/examples/txt2img/multi.py

StreamDiffusionWrapper#txt2img()

0.19

/utils/wrapper.py

main()

0.19

/examples/txt2img/multi.py

main()

0.19

/examples/optimal-performance/single.py

__init__

0.19

/src/streamdiffusion/image_filter.py

Engine#infer()

0.19

/src/streamdiffusion/acceleration/tensorrt/utilities.py

image_generation_process()

0.19

/examples/optimal-performance/single.py

image_generation_process()

0.19

/examples/optimal-performance/multi.py

__init__

0.19

/examples/optimal-performance/single.py

__init__

0.18

/examples/txt2img/single.py

StreamDiffusion#unet_step()

0.18

/src/streamdiffusion/pipeline.py

UNet2DConditionModelEngine

0.18

/src/streamdiffusion/acceleration/tensorrt/engine.py

UNet2DConditionModelEngine#__call__()

0.18

/src/streamdiffusion/acceleration/tensorrt/engine.py

BaseModel#get_minmax_dims()

0.18

/src/streamdiffusion/acceleration/tensorrt/models.py

main()

0.18

/examples/optimal-performance/multi.py

Api#_predict()

0.18

/demo/realtime-txt2img/server/main.py

AutoencoderKLEngine#__init__()

0.18

/src/streamdiffusion/acceleration/tensorrt/engine.py

BaseModel#__init__()

0.18

/src/streamdiffusion/acceleration/tensorrt/models.py

__init__

0.17

/utils/wrapper.py

run()

0.17

/examples/benchmark/multi.py

run()

0.17

/examples/benchmark/single.py

PredictResponseModel

0.17

/demo/realtime-txt2img/server/main.py

main()

0.17

/examples/txt2img/single.py

BaseModel#get_model()

0.17

/src/streamdiffusion/acceleration/tensorrt/models.py

UNet#get_shape_dict()

0.17

/src/streamdiffusion/acceleration/tensorrt/models.py

__init__

0.16

/src/streamdiffusion/acceleration/tensorrt/__init__.py

__init__

0.16

/src/streamdiffusion/acceleration/tensorrt/builder.py

StreamDiffusion#enable_similar_image_filter()

0.16

/src/streamdiffusion/pipeline.py

__init__

0.16

/examples/benchmark/single.py

BaseModel#get_input_profile()

0.16

/src/streamdiffusion/acceleration/tensorrt/models.py

StreamDiffusion#prepare()

0.16

/src/streamdiffusion/pipeline.py

__init__

0.16

/demo/realtime-txt2img/server/main.py

StreamDiffusion#__call__()

0.15

/src/streamdiffusion/pipeline.py

VAEEncoder#__init__()

0.15

/src/streamdiffusion/acceleration/tensorrt/models.py

BaseModel#get_sample_input()

0.15

/src/streamdiffusion/acceleration/tensorrt/models.py

VAE#get_sample_input()

0.15

/src/streamdiffusion/acceleration/tensorrt/models.py

Get CodeMuse in your IDE.

Get for VS Code — It's free

Star on GitHub

Spend less time
onboarding a repository

Get CodeMuse in your IDE.

Features

Search

Automated documentation

Key-up assistant

Code implementation

Get informed about CodeMuse

Spend less time (()=>{var e=async t=>{await(await t())()};(self.Astro||(self.Astro={})).load=e;window.dispatchEvent(new Event("astro:load"));})();onboarding a repository

Get CodeMuse in your IDE.

Features

Search

Automated documentation

Key-up assistant

Code implementation

Get informed about CodeMuse

Spend less time
onboarding a repository