PictSure 1.0
Collection
5 items • Updated
This model combines a dinov2-large encoder with a custom transformer classifier.
import torch
from huggingface_hub import hf_hub_download
from safetensors.torch import load_file
# Download the model
model_path = hf_hub_download(repo_id="pictsure/PictSure-1.5-large", filename="model.safetensors")
# Load the model state
state_dict = load_file(model_path)
# ... initialize your model architecture and load state_dict