neurovlm.models.load_model

Contents

neurovlm.models.load_model#

neurovlm.models.load_model(name)[source]#

Alias to .from_pretrained methods in model classes.

Parameters:

name (str, {"proj_head_text_infonce", "proj_head_image_infonce", "proj_head_text_mse", "autoencoder", "specter"}) – Name of model.

Return type:

model