mirror of
https://github.com/Niko1221/Strata.git
synced 2026-11-07 14:21:06 -10:00
Both Unsloth DeepSeek sets are offered on a DGX Spark: both passed a real Spark run (AARCH64_NOT_YET is empty); tests keep the mechanism
This commit is contained in:
1 parent
1a26492064
commit
38f56619cc
2 files changed
+17
-3
No files matched your search
@@ -618,8 +618,9 @@ MAC_SET_PACK = {"IST-DAS-2.00bpw": "IQ2_XXS", "IST-DAS-2.25bpw": "IQ2_XS", "UD-I
|
||||
# Remove a name here when its set has run on a Mac.
|
||||
MAC_NOT_YET = ("UD-IQ1_M",)
|
||||
MAC_NOT_YET_LINE = "Unsloth's UD-IQ1_M set doesn't run on a Mac yet; the UD-IQ1_S, 2.00 and 2.25 sets do."
|
||||
# The same two sets were never run on a DGX Spark (linux-aarch64, GB10) either: kept off it the same way until a Spark run passes (remove the names then).
|
||||
AARCH64_NOT_YET = ("UD-IQ1_S", "UD-IQ1_M")
|
||||
# The same two sets were kept off a DGX Spark (linux-aarch64, GB10) until a Spark run passed: both passed there on 2026-10-30 (real `strata launch --model-path`,
|
||||
# 10 of 10 answers, tool call correct), so the tuple is empty. Put a name back here to hide a set from the Spark again.
|
||||
AARCH64_NOT_YET = ()
|
||||
AARCH64_NOT_YET_LINE = "Unsloth's 1-bit DeepSeek sets were not tested on a DGX Spark yet; the 2.00 and 2.25 sets were."
|
||||
|
||||
|
||||
|
||||
@@ -22,7 +22,7 @@ MAC = {**HW_12, "mac": True, "ram_gb": 128.0, "disk_free_gb": 900.0, "gpus": [],
|
||||
PC = {**HW_12, "ram_gb": 64.0, "disk_free_gb": 900.0, "engine_vendor": "nvidia"}
|
||||
UD = ("UD-IQ1_M",) # the sets kept off a Mac
|
||||
RUNS = "UD-IQ1_S" # the one that was run there
|
||||
BOTH = ("UD-IQ1_S", "UD-IQ1_M") # both are still kept off a DGX Spark
|
||||
BOTH = ("UD-IQ1_S", "UD-IQ1_M") # both ran on a DGX Spark (10-30): AARCH64_NOT_YET is empty; SparkGate patches it back to test the mechanism
|
||||
|
||||
|
||||
def ud_files(model):
|
||||
@@ -213,6 +213,9 @@ class SparkGate(unittest.TestCase):
|
||||
p = mock.patch.object(Catalog, "hardware", lambda s, fresh=False: dict(self.hw))
|
||||
p.start()
|
||||
self.addCleanup(p.stop)
|
||||
q = mock.patch.object(dsv41_fit, "AARCH64_NOT_YET", BOTH) # the mechanism, as it was while the sets were kept off
|
||||
q.start()
|
||||
self.addCleanup(q.stop)
|
||||
|
||||
def test_the_rule(self):
|
||||
self.assertEqual(dsv41_fit.AARCH64_NOT_YET, BOTH)
|
||||
@@ -272,5 +275,15 @@ class SparkGate(unittest.TestCase):
|
||||
self.assertIn("nothing has been downloaded", text)
|
||||
|
||||
|
||||
class SparkOffered(unittest.TestCase):
|
||||
"""Both Unsloth sets ran on a DGX Spark (2026-10-30): the production rule hides nothing there."""
|
||||
|
||||
def test_nothing_is_hidden_on_a_spark(self):
|
||||
self.assertEqual(dsv41_fit.AARCH64_NOT_YET, ())
|
||||
with spark():
|
||||
for m in BOTH + ("IST-DAS-2.25bpw",):
|
||||
self.assertIsNone(dsv41_fit.not_here(m))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in new issue
Block a user