87a0ba3db3
* export clip models * export to hf refactored export code * export mclip, general refactoring cleanup * updated conda deps * do transforms with pillow and numpy, add tokenization config to export, general refactoring * moved conda dockerfile, re-added poetry * minor fixes * updated link * updated tests * removed `requirements.txt` from workflow * fixed mimalloc path * removed torchvision * cleaner np typing * review suggestions * update default model name * update test
15 lines
439 B
Python
15 lines
439 B
Python
import json
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
|
|
def get_model_path(output_dir: Path | str) -> Path:
|
|
output_dir = Path(output_dir)
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|
return output_dir / "model.onnx"
|
|
|
|
|
|
def save_config(config: Any, output_path: Path | str) -> None:
|
|
output_path = Path(output_path)
|
|
output_path.parent.mkdir(parents=True, exist_ok=True)
|
|
json.dump(config, output_path.open("w"))
|