Function save_clip_encoder

python/clip.py:25–29

(clip_encoder, path)

    save_layer_norm(clip_encoder_layer.layer_norm2, pathlib.Path(path, 'mlp_ln'))

def save_clip_encoder(clip_encoder, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)
    for i, layer in enumerate(clip_encoder.layers):
        save_clip_encoder_layer(layer, pathlib.Path(path, f'blocks/{i}'))
    save_scalar(len(clip_encoder.layers), "n_layer", path)

def save_clip_text_embeddings(clip_text_embeddings, path):
    pathlib.Path(path).mkdir(parents=True, exist_ok=True)