Add profile presets and min-p controls, embeddings and VoxCPM workers
This commit is contained in:
1 parent
74dd2ed8ba
commit
a7b30b62d2
32 files changed
+615
-64
No files matched your search
@@ -7,6 +7,17 @@ from unittest.mock import patch
|
||||
from catalog import Catalog, safe_url, repo_id, is_derived_model, runtime_metadata
|
||||
|
||||
class CatalogTests(unittest.TestCase):
|
||||
def test_embeddings_search_has_own_hub_category(self):
|
||||
from urllib.parse import urlsplit,parse_qs
|
||||
from profiles import SCHEMAS
|
||||
from execution_setup import assess
|
||||
with tempfile.TemporaryDirectory() as d,patch('catalog.metadata',return_value=[]) as fetch:
|
||||
Catalog(d).search('embedding','embeddings')
|
||||
self.assertEqual(parse_qs(urlsplit(fetch.call_args.args[0]).query)['filter'],['feature-extraction'])
|
||||
self.assertIn('embeddings',SCHEMAS)
|
||||
result=assess(dict(kind='embeddings',repo='owner/model',file='model.gguf'),{})
|
||||
self.assertNotEqual(result['state'],'ready')
|
||||
|
||||
def test_bounded_hub_runtime_metadata(self):
|
||||
hint=runtime_metadata(dict(pipeline_tag='text-to-image',library_name='diffusers',config={'architectures':['NewImagePipeline']},tags=['image-to-image']))
|
||||
self.assertEqual(hint['architecture'],'NewImagePipeline')
|
||||
|
||||
Reference in new issue
Block a user