From 2f3b38b518f36c092362a1dd0df8d1e97d338332 Mon Sep 17 00:00:00 2001 From: Vladimir Loginov Date: Wed, 18 Feb 2026 13:41:14 +0300 Subject: [PATCH] Fix pretrain model paths & update requirements --- configs/models/deepgen.py | 4 ++-- configs/models/deepgen_scb.py | 4 ++-- requirements.txt | 1 + 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/configs/models/deepgen.py b/configs/models/deepgen.py index adefe61..4f4b692 100644 --- a/configs/models/deepgen.py +++ b/configs/models/deepgen.py @@ -5,8 +5,8 @@ from src.models.sd3_kontext.transformer_sd3_dynamic import SD3Transformer2DModel -sd3_5_model_name_or_path = "model_zoo/UniPic2-SD3.5M-Kontext-2B" -qwen2_5_vl_model_name_or_path = "model_zoo/Qwen2.5-VL-3B-Instruct" +sd3_5_model_name_or_path = "Skywork/UniPic2-SD3.5M-Kontext-2B" +qwen2_5_vl_model_name_or_path = "Qwen/Qwen2.5-VL-3B-Instruct" tokenizer = dict( type=AutoTokenizer.from_pretrained, diff --git a/configs/models/deepgen_scb.py b/configs/models/deepgen_scb.py index c9294de..053a07b 100644 --- a/configs/models/deepgen_scb.py +++ b/configs/models/deepgen_scb.py @@ -5,8 +5,8 @@ from src.models.sd3_kontext.transformer_sd3_dynamic import SD3Transformer2DModel -sd3_5_model_name_or_path = "model_zoo/UniPic2-SD3.5M-Kontext-2B" -qwen2_5_vl_model_name_or_path = "model_zoo/Qwen2.5-VL-3B-Instruct" +sd3_5_model_name_or_path = "Skywork/UniPic2-SD3.5M-Kontext-2B" +qwen2_5_vl_model_name_or_path = "Qwen/Qwen2.5-VL-3B-Instruct" tokenizer = dict( type=AutoTokenizer.from_pretrained, diff --git a/requirements.txt b/requirements.txt index 585d0f9..93bcfca 100644 --- a/requirements.txt +++ b/requirements.txt @@ -37,6 +37,7 @@ huggingface-hub==0.34.4 idna==3.10 imageio==2.37.0 importlib_metadata==8.6.1 +inflect==7.5.0 ipykernel==6.29.5 ipython==9.2.0 ipython_pygments_lexers==1.1.1