diff --git a/.gitignore b/.gitignore index 69e4a3a39..4b0422cd9 100644 --- a/.gitignore +++ b/.gitignore @@ -32,7 +32,7 @@ models/* /result /perplexity /embedding -/finetune +/train-text-from-scratch /benchmark-matmult /vdot /Pipfile diff --git a/Makefile b/Makefile index acf3dca79..17687f4fb 100644 --- a/Makefile +++ b/Makefile @@ -1,5 +1,5 @@ # Define the default target now so that it is always the first target -BUILD_TARGETS = main quantize quantize-stats perplexity embedding vdot finetune +BUILD_TARGETS = main quantize quantize-stats perplexity embedding vdot train-text-from-scratch ifdef LLAMA_BUILD_SERVER BUILD_TARGETS += server @@ -289,7 +289,7 @@ save-load-state: examples/save-load-state/save-load-state.cpp build-info.h ggml. server: examples/server/server.cpp examples/server/httplib.h examples/server/json.hpp build-info.h ggml.o llama.o common.o $(OBJS) $(CXX) $(CXXFLAGS) -Iexamples/server $(filter-out %.h,$(filter-out %.hpp,$^)) -o $@ $(LDFLAGS) -finetune: examples/train-text-from-scratch/train-text-from-scratch.cpp build-info.h ggml.o llama.o $(OBJS) +train-text-from-scratch: examples/train-text-from-scratch/train-text-from-scratch.cpp build-info.h ggml.o llama.o $(OBJS) $(CXX) $(CXXFLAGS) $(filter-out %.h,$^) -o $@ $(LDFLAGS) build-info.h: $(wildcard .git/index) scripts/build-info.sh diff --git a/examples/train-text-from-scratch/README.md b/examples/train-text-from-scratch/README.md index 02200e68d..c51d01604 100644 --- a/examples/train-text-from-scratch/README.md +++ b/examples/train-text-from-scratch/README.md @@ -7,7 +7,7 @@ Basic usage instructions: wget https://github.com/brunoklein99/deep-learning-notes/blob/master/shakespeare.txt # train -../../finetune \ +../../train-text-from-scratch \ --vocab-model ../models/ggml-vocab.bin \ --ctx 64 --embd 256 --head 8 --layer 16 \ --checkpoint-in chk-shakespeare-256x16.bin \