Upload 9 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/reference.png filter=lfs diff=lfs merge=lfs -text
+examples/source.png filter=lfs diff=lfs merge=lfs -text

inference.py

@@ -18,7 +18,7 @@ Example:
     --source examples/source.png \
     --reference examples/reference.png \
     --output output.png \
-    --repo_id ChengYou305/DI2FIX
+    --repo_id ChengYou305/DI2FIX_HF
 """
 
 import argparse
@@ -41,7 +41,7 @@ from di2fix_utils import (
 
 
 DEFAULT_CONFIG: Dict[str, Any] = {
-    "model_repo_id": "ChengYou305/DI2FIX",
+    "model_repo_id": "ChengYou305/DI2FIX_HF",
     "checkpoint_filename": "model_80001.pkl",
     "base_difix_repo_id": "nvidia/difix_ref",
     "lora_rank_vae": 4,