diff --git a/.gitea/embed.yaml b/.gitea/embed.yaml new file mode 100644 index 0000000..22c5bda --- /dev/null +++ b/.gitea/embed.yaml @@ -0,0 +1,37 @@ +name: Run VectorLoader Script + +on: + push: + branches: + - main + +jobs: + Explore-Gitea-Actions: + runs-on: ubuntu-latest + container: catthehacker/ubuntu:act-latest + steps: + - name: Check out repository code + uses: actions/checkout@v4 + + - name: Set up Python + uses: actions/setup-python@v3 + with: + python-version: '3.11.7' + + - name: Clone additional repository + run: | + git config --global credential.helper cache + git clone https://fabel:${{ secrets.CICD }}@gitea.fabelous.app/fabel/VectorLoader.git + + - name: Install dependencies + run: | + cd VectorLoader + python -m pip install --upgrade pip + pip install -r requirements.txt + + - name: Run vectorizing + env: + VECTORDB_TOKEN: ${{ secrets.VECTORDB_TOKEN }} + run: | + cd VectorLoader + python -m src.run --full diff --git a/.gitea/test.yaml b/.gitea/test.yaml new file mode 100644 index 0000000..b754440 --- /dev/null +++ b/.gitea/test.yaml @@ -0,0 +1,37 @@ +name: Gitea Actions For AIIA +run-name: ${{ gitea.actor }} is testing out Gitea Actions 🚀 +on: [push] + +jobs: + Explore-Gitea-Actions: + runs-on: ubuntu-latest + container: catthehacker/ubuntu:act-latest + steps: + - name: Check out repository code + uses: actions/checkout@v4 + + - name: Set up Python + uses: actions/setup-python@v5 + with: + python-version: '3.11.7' + + - name: Cache pip and model + uses: actions/cache@v3 + with: + path: | + ~/.cache/pip + ./fabel + key: ${{ runner.os }}-pip-model-${{ hashFiles('requirements.txt', 'requirements-dev.txt') }} + + restore-keys: | + ${{ runner.os }}-pip-model- + + - name: Install dependencies + run: | + python -m pip install --upgrade pip + pip install -r requirements.txt + pip install -r requirements-dev.txt + pip install -e . + - name: Run tests + run: | + pytest tests/ \ No newline at end of file diff --git a/src/aiunn/upsampler/aiunn.py b/src/aiunn/upsampler/aiunn.py index b0c26b2..ecb3c21 100644 --- a/src/aiunn/upsampler/aiunn.py +++ b/src/aiunn/upsampler/aiunn.py @@ -149,8 +149,8 @@ if __name__ == "__main__": upsampler = aiuNN(base_model, config=ai_config) # Save the model (both configuration and weights). - upsampler.save("hehe") + upsampler.save("aiunn") # Now load using the overridden load method; this will load the complete model. - upsampler_loaded = aiuNN.load("hehe", precision="bf16") + upsampler_loaded = aiuNN.load("aiunn", precision="bf16") print("Updated configuration:", upsampler_loaded.config.__dict__)