WorldDisasterLM-8B / scripts /export_gguf.py
drdeveloper88's picture
Upload WorldDisasterLM-8B source code: FastAPI backend, training pipeline, 11-language support
495526b
Raw
History Blame Contribute Delete
575 Bytes
import argparse
from worlddisasterlm.optimization.export_gguf import export_gguf
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser(description="Export WorldDisasterLM-8B to GGUF")
parser.add_argument("--model-path", default="checkpoints/worlddisasterlm-8b-qlora")
parser.add_argument("--output-path", default="artifacts/worlddisasterlm-8b.gguf")
return parser.parse_args()
def main() -> None:
args = parse_args()
export_gguf(model_path=args.model_path, output_path=args.output_path)
if __name__ == "__main__":
main()