fix: check for wte.weight along with embed_tokens.weight (#356) #14

	name: Image
	on:
	push:
	branches: [ "main" ]
	pull_request:
	branches: [ "main" ]

	jobs:
	build:
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v4
	- name: Free disk space
	run: \|
	sudo swapoff -a
	sudo rm -f /swapfile
	sudo apt clean
	docker rmi $(docker image ls -aq)
	df -h
	- name: Build image
	run: \|
	docker build -t fms-hf-tuning:dev . -f build/Dockerfile

Provide feedback