partarstu / transformers-in-java

Experimental project for AI and NLP based on Transformer Architecture
https://partarstu.github.io/transformers-in-java/
Apache License 2.0
10 stars 2 forks source link

First set of improvements #2

Closed partarstu closed 10 months ago

partarstu commented 10 months ago
  1. Changed the version to 1.0-SNAPSHOT
  2. Added README file explaining how the training could be done and how to build the docker image
  3. Upgraded to JDK 21
  4. Added IntelliJ run configs for generative (GPT-like) and Q&A models
  5. Added docker-based fixes for https://github.com/partarstu/transformers-in-java/issues/1
  6. Added memory-related fixes.
  7. Upgraded some libs (in order to get some critical vulnerabilities fixed)