back to run

eljandoubi · Aug 24, 2024 · 869ef45 · 869ef45
1 parent 914aa6f
commit 869ef45
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.github/workflows/python-package-conda.yml b/.github/workflows/python-package-conda.yml
@@ -21,6 +21,11 @@ jobs:
     - name: Build
       run: |
         make build
+    - name: Run
+      run: |
+        make run
+      env:
+        HF_TOKEN: ${{ secrets.HF_TOKEN }}
     - name: Clean
       run: |
         make clean
diff --git a/src/utils/model_loader.py b/src/utils/model_loader.py
@@ -38,6 +38,7 @@ def load_hf_model(model_id: str,
                 tensors[key] = f.get_tensor(key)
 
     # Load the model's config
+    print("Load the model's config")
     with open(os.path.join(model_path, "config.json"), "r", encoding="utf-8") as f:
         model_config_file = json.load(f)
         config = PaliGemmaConfig.from_dict(model_config_file)