Workaround for nvidia issue where VAE uses 3x more memory on torch 2.… #1376

	name: Unit Tests

	on:
	push:
	branches: [ main, master ]
	pull_request:
	branches: [ main, master ]

	jobs:
	test:
	strategy:
	matrix:
	os: [ubuntu-latest, windows-2022, macos-latest]
	runs-on: ${{ matrix.os }}
	continue-on-error: true
	steps:
	- uses: actions/checkout@v4
	- name: Set up Python
	uses: actions/setup-python@v4
	with:
	python-version: '3.12'
	- name: Install requirements
	run: \|
	python -m pip install --upgrade pip
	pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
	pip install -r requirements.txt
	- name: Run Unit Tests
	run: \|
	pip install -r tests-unit/requirements.txt
	python -m pytest tests-unit

Provide feedback