Spaces:
Restarting
on
CPU Upgrade
Restarting
on
CPU Upgrade
update cache
Browse files
src/submission/check_validity.py
CHANGED
@@ -36,6 +36,8 @@ memory = Memory('is_model_on_hub_cache', verbose=0)
|
|
36 |
|
37 |
@memory.cache
|
38 |
def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_remote_code=False, test_tokenizer=False) -> tuple[bool, str]:
|
|
|
|
|
39 |
if 'Bielik-11' in model_name:
|
40 |
return True, "Bielik", None
|
41 |
try:
|
|
|
36 |
|
37 |
@memory.cache
|
38 |
def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_remote_code=False, test_tokenizer=False) -> tuple[bool, str]:
|
39 |
+
if 'Bielik-11B-v2.2-M' in model_name:
|
40 |
+
return False, "priv Bielik", None
|
41 |
if 'Bielik-11' in model_name:
|
42 |
return True, "Bielik", None
|
43 |
try:
|