Spaces:
Restarting
on
CPU Upgrade
Restarting
on
CPU Upgrade
Update src/submission/check_validity.py
Browse files
src/submission/check_validity.py
CHANGED
@@ -36,8 +36,10 @@ 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 '
|
40 |
-
return False, "
|
|
|
|
|
41 |
try:
|
42 |
config = AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=trust_remote_code, token=token)
|
43 |
if test_tokenizer:
|
|
|
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:
|
44 |
config = AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=trust_remote_code, token=token)
|
45 |
if test_tokenizer:
|