diff --git a/.github/workflows/python-package-conda.yml b/.github/workflows/python-package-conda.yml index 6daa9d8..d0b6f28 100644 --- 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 index 5ba79b5..4f809a7 100644 --- 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)