Fabio
c36510c21e
Some checks failed
Linux Build / Linux cpu (push) Has been cancelled
Linux Build / Linux cpu-clang (push) Has been cancelled
Linux Build / Linux vulkan (push) Has been cancelled
mac-build / macOS CPU (push) Has been cancelled
Nix Build / Nix (cpu) (push) Has been cancelled
Nix Build / Nix (vulkan) (push) Has been cancelled
windows-build / Windows CPU (push) Has been cancelled
51 lines
2.5 KiB
Python
51 lines
2.5 KiB
Python
"""Standard Bark source identity and shared-converter isolation checks."""
|
|
from dataclasses import FrozenInstanceError
|
|
import json
|
|
from pathlib import Path
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
REPO = Path(__file__).resolve().parents[2]
|
|
sys.path.insert(0, str(REPO / "tools/community_models"))
|
|
import convert_underdog_bark3_plus as packed
|
|
import convert_underdog_bark3_standard as standard
|
|
|
|
|
|
class BarkStandardConversionTest(unittest.TestCase):
|
|
def test_profiles_are_immutable_and_do_not_contaminate_plus(self):
|
|
original = packed.package_spec(REPO)
|
|
spec = packed.package_spec(REPO, standard.STANDARD_PROFILE)
|
|
self.assertEqual(spec["display_name"], "Underdog Bark 3.0")
|
|
self.assertEqual(spec["family"], "parakeet_tdt")
|
|
self.assertEqual(spec["languages"], ["en"])
|
|
self.assertEqual(spec["packages"], [])
|
|
self.assertEqual(spec["ui"]["docs"], ["docs/models/underdog_bark3.md"])
|
|
self.assertEqual(packed.package_spec(REPO), original)
|
|
self.assertEqual(packed.VARIANT, "underdog_bark3_plus")
|
|
self.assertEqual(packed.PLUS_PROFILE.bf16_f32_prefixes, ())
|
|
self.assertTrue(standard.STANDARD_PROFILE.bf16_f32_prefixes)
|
|
with self.assertRaises(FrozenInstanceError):
|
|
standard.STANDARD_PROFILE.variant = "underdog_bark3_plus"
|
|
|
|
def test_cross_model_staging_is_rejected_before_packing(self):
|
|
for profile, other in [(standard.STANDARD_PROFILE, packed.PLUS_PROFILE),
|
|
(packed.PLUS_PROFILE, standard.STANDARD_PROFILE)]:
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
staging = Path(directory)
|
|
metadata = dict(source=other.source_repo, revision=other.source_revision,
|
|
model_sha256=other.source_sha256, variant=other.variant)
|
|
(staging / "conversion.json").write_text(json.dumps(metadata))
|
|
with self.assertRaisesRegex(ValueError, "different source/variant"):
|
|
packed.pack(staging, staging / "missing-tool", staging / "out.gguf", "mixed-f16", profile=profile)
|
|
|
|
def test_unpinned_standard_checkpoint_is_rejected(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = Path(directory)
|
|
(root / "model.bark").write_bytes(b"wrong checkpoint")
|
|
with self.assertRaisesRegex(ValueError, "pinned revision"):
|
|
packed.prepare(root, root / "staging", standard.STANDARD_PROFILE)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|