michael/Athena-Deck
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Classify library components consistently and hide standalone profile action

Browse Source
This commit is contained in:
Mikei386
2026-09-28 17:41:06 +02:00
parent 057b9df644
commit 413906f619
6 changed files with 37 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_model_management.py
+10
View File
@@ -73,3 +73,13 @@ class ManagementTests(unittest.TestCase):
with self.assertRaises(ValueError):restarted.assign(dict(id=saved['id'],revision=2,components=ids))
request=self.request();request['model_id']=ids['vae']
with self.assertRaises(ValueError):restarted.save(request)
def test_auxiliary_files_cannot_be_profile_models(self):
from catalog import file_role
for name,role in [('text_encoders/qwen3vl_8b_int8_convrot.safetensors','text_encoder'),('vae/qwen_image_2.1_vae_bf16.safetensors','vae'),('qwen_image_vae.safetensors','vae'),('mmproj-model.gguf','auxiliary')]:
with self.subTest(name=name):
self.assertEqual(file_role(name)['role'],role)
path=self.root/'models'/self.model_id/'entry.json';data=json.loads(path.read_text());data['file']=name;path.write_text(json.dumps(data))
suffix=Path(name).suffix;(path.parent/('model'+suffix)).write_bytes(b'GGUFtest')
self.assertFalse(self.catalog.status()['entries'][0]['profile_eligible'])
with self.assertRaises(ValueError):self.profiles.save(self.request())
self.assertTrue(file_role('qwen-image-2.1-UC-Q8_0.gguf')['profile_eligible'])