fixed weight download

Makefile

@@ -63,11 +63,14 @@ cov:
 	$(CONDA_RUN) pytest --cov=mini_transformer --cov-report=term-missing
 
 fetch-test:
-	$(CONDA_RUN) mini-transformer-fetch AlaBoussoffara/transformer_test --name test_model_v1 --force
+	mini-transformer-fetch AlaBoussoffara/transformer_test --force
+	cp -rf ./trained_models/AlaBoussoffara__transformer_test/test_model_v1 ./trained_models/test_model_v1
+	rm -rf ./trained_models/AlaBoussoffara__transformer_test
 
 fetch-small:
-	$(CONDA_RUN) mini-transformer-fetch AlaBoussoffara/transformer_small --name small_model_v1 --force
-
+	mini-transformer-fetch AlaBoussoffara/transformer_small --force
+	cp -rf ./trained_models/AlaBoussoffara__transformer_small/small_model_v1 ./trained_models/small_model_v1
+	rm -rf ./trained_models/AlaBoussoffara__transformer_small/
 infer:
 	$(CONDA_RUN) mini-transformer-infer --model small_model_v1 -t "translate this sentence."