Txu647 commited on
Commit
c2f3dfc
·
1 Parent(s): ec5b3cb

fix: update checkpoint path to TSXu/Unicalli_Pro

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. inference.py +2 -2
app.py CHANGED
@@ -68,7 +68,7 @@ def init_generator():
68
  device="cuda",
69
  offload=False,
70
  intern_vlm_path="OpenGVLab/InternVL3-1B",
71
- checkpoint_path="TSXu/UniCalli-pro",
72
  font_descriptions_path='dataset/chirography.json',
73
  author_descriptions_path='dataset/calligraphy_styles_en.json',
74
  use_deepspeed=False,
 
68
  device="cuda",
69
  offload=False,
70
  intern_vlm_path="OpenGVLab/InternVL3-1B",
71
+ checkpoint_path="TSXu/Unicalli_Pro",
72
  font_descriptions_path='dataset/chirography.json',
73
  author_descriptions_path='dataset/calligraphy_styles_en.json',
74
  use_deepspeed=False,
inference.py CHANGED
@@ -20,8 +20,8 @@ from src.flux.xflux_pipeline import XFluxSampler
20
 
21
 
22
  # HuggingFace Hub model IDs
23
- HF_MODEL_ID = "TSXu/UniCalli-base"
24
- HF_CHECKPOINT_FILENAME = "unicalli-base_cleaned.bin"
25
  HF_INTERNVL_ID = "OpenGVLab/InternVL3-1B"
26
 
27
 
 
20
 
21
 
22
  # HuggingFace Hub model IDs
23
+ HF_MODEL_ID = "TSXu/Unicalli_Pro"
24
+ HF_CHECKPOINT_FILENAME = "unicalli_pro_chars7.bin"
25
  HF_INTERNVL_ID = "OpenGVLab/InternVL3-1B"
26
 
27