File size: 420 Bytes
704fa80 1a9eb8a 704fa80 1a9eb8a | 1 2 3 4 5 6 7 8 9 | from pathlib import Path
from semantic_model import SemanticColorizer, load_semantic, save_semantic
def load_model(source, revision=None, device="cpu"):
from huggingface_hub import snapshot_download
path=Path(source)
if not path.is_dir(): path=Path(snapshot_download(source,revision=revision,allow_patterns=["config.json","model.safetensors"]))
return load_semantic(path,device)
save_model=save_semantic
|