Commit graph

12 commits

Author SHA1 Message Date
KKKKKKKevin
7f361aedf4
better symbol (#344) 2025-05-13 14:22:47 +08:00
KKKKKKKevin
34d43290e0
Feat/fix no llama.cpp (#297)
* feat: what? no llama.cpp

* add cache
2025-04-26 15:26:59 +08:00
Zachary Pitroda
053090937d
Added CUDA support (#228)
* Add CUDA support

- CUDA detection
- Memory handling
- Ollama model release after training

* Fix logging issue

added cuda support flag so log accurately reflected cuda toggle

* Update llama.cpp rebuild

Changed llama.cpp to only check if cuda support is enabled and if so rebuild during the first build rather than each run

* Improved vram management

Enabled memory pinning and optimizer state offload

* Fix CUDA check

rewrote llama.cpp rebuild logic, added manual y/n toggle if user wants to enable cuda support

* Added fast restart and fixed CUDA check command

Added make docker-restart-backend-fast to restart the backend and reflect code changes without causing a full llama.cpp rebuild

Fixed make docker-check-cuda command to correctly reflect cuda support

* Added docker-compose.gpu.yml

Added docker-compose.gpu.yml to fix error on machines without nvidia gpu and made sure "\n" is added before .env modification

* Fixed cuda toggle

Last push accidentally broke cuda toggle

* Code review fixes

Fixed errors resulting from removed code:
- Added return save_path to end of save_hf_model function
- Rolled back download_file_with_progress function

* Update Makefile

Use cuda by default when using docker-restart-backend-fast

* Minor cleanup

Removed unnecessary makefile command and fixed gpu logging

* Delete .gpu_selected

* Simplified cuda training code

- Removed dtype setting to let torch automatically handle it
- Removed vram logging
- Removed Unnecessary/old comments

* Fixed gpu/cpu selection

Made "make docker-use-gpu/cpu" command work with .gpu_selected flag and changed "make docker-restart-backend-fast" command to respect flag instead of always using gpu

* Fix Ollama embedding error

Added custom exception class for Ollama embeddings, which seemed to be returning keyword arguments while the Python exception class only accepts positional ones

* Fixed model selection & memory error

Fixed training defaulting to 0.5B model regardless of selection and fixed "free(): double free detected in tcache 2" error caused by cuda flag being passed incorrectly
2025-04-25 10:20:36 +08:00
KKKKKKKevin
0e442ed17c
Fix/make meta data field optional (#237)
* fix:metadata to optional

* docker command check

* add exception stack
2025-04-15 19:51:33 +08:00
KKKKKKKevin
a85f53af7d
Feat/windows asc ii (#221)
* feat: win ascii fit

* fix for windows

* fix windows log

* fix windows logo
2025-04-14 14:49:02 +08:00
Zeeland
5c085dbbbf
feat: update Docker Compose command for Windows compatibility (#181) 2025-04-14 10:02:21 +08:00
KKKKKKKevin
4c71dce682
Feat: Remove conda from setup (#195)
* feat:remove conda from setup

* cool. more code about removing conda & env

* optimize script

* add python check

* optimize setup

* optimize setup

* add warning info

* optimize check

* add node check & npm check

* add cmake install suggestion

* add poetry suggestion

* add python command check

* add python tools

* add poetry

* add graphrag poetry install command optimization

* active poetry shell

* fix duplicated log

* avoid poetry multipul active

* add sqlite3 check

* optimize sqlite3 setup suggestion

* optimize

* add gitignore

* reverse compose detect
2025-04-10 19:39:26 +08:00
Chuan Trump
778c379636
feat: Adapting the docker compose command (#168) 2025-04-08 19:47:18 +08:00
weiyong1024
4a019c4ec5
adjust env variable configuration for Apple Silicon (#159) 2025-04-07 11:54:35 +08:00
GoForceX
9a914c1529 fix: Invalid progress file path on Windows (#145) 2025-04-03 10:35:01 +08:00
KKKKKKKevin
f64ddc108b
feat: Add docker support (#99)
* feat:Add Docker Support

* translate
2025-03-30 15:15:32 +08:00
Kevin
7f7d64210e Initial commit 2025-03-20 00:37:54 +08:00