apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Inject num2words function instead of module into transformers processing_smolvlm

Browse source
This commit is contained in:
Sarvesh 2026-06-09 16:28:32 +05:30
parent e845504d7f
commit 70e60aa6aa
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
demo.ipynb
View file

@ -190,14 +190,14 @@
"import importlib.util\n",
"if importlib.util.find_spec(\"num2words\") is not None:\n",
" try:\n",
" import num2words\n",
" from num2words import num2words as num2words_func\n",
" import transformers.utils.import_utils\n",
" transformers.utils.import_utils._num2words_available = True\n",
" for mod_name in list(sys.modules.keys()):\n",
" if \"processing_smolvlm\" in mod_name or \"smolvlm\" in mod_name:\n",
" mod = sys.modules[mod_name]\n",
" if hasattr(mod, \"num2words\") and getattr(mod, \"num2words\") is None:\n",
" setattr(mod, \"num2words\", num2words)\n",
" if hasattr(mod, \"num2words\"):\n",
" setattr(mod, \"num2words\", num2words_func)\n",
" except Exception:\n",
" pass\n"
]
@ -449,19 +449,20 @@
"# 1. Initialize SmolVLM2 base model, processor, and tokenizer\n",
"print(\"Loading SmolVLM2 model (2.2B-Instruct)...\")\n",
"\n",
"\n",
"# Force-refresh Hugging Face dependency cache for num2words if installed mid-session\n",
"import sys\n",
"import importlib.util\n",
"if importlib.util.find_spec(\"num2words\") is not None:\n",
" try:\n",
" import num2words\n",
" from num2words import num2words as num2words_func\n",
" import transformers.utils.import_utils\n",
" transformers.utils.import_utils._num2words_available = True\n",
" for mod_name in list(sys.modules.keys()):\n",
" if \"processing_smolvlm\" in mod_name or \"smolvlm\" in mod_name:\n",
" mod = sys.modules[mod_name]\n",
" if hasattr(mod, \"num2words\") and getattr(mod, \"num2words\") is None:\n",
" setattr(mod, \"num2words\", num2words)\n",
" if hasattr(mod, \"num2words\"):\n",
" setattr(mod, \"num2words\", num2words_func)\n",
" except Exception:\n",
" pass\n",
"\n",