Commit graph

7 commits

Author SHA1 Message Date
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
aa38f672f0
fix: shade -> shade_merge_info (#265)
* fix: shade -> shade_merge_info

* add convert array
2025-04-22 15:05:08 +08:00
doubleBlack2
1fb4dd9ebb
Change topics_generator timeout to 30 (#263) 2025-04-22 11:57:45 +08:00
Airmomo
28966cc307
fix: implement dynamic top_p parameter adjustment for LLM API calls (#104)
- Add _top_p_adjusted flag to track parameter adjustments
    - Implement _fix_top_p_param to adjust invalid top_p values
    - Enhance _call_llm_with_retry with parameter adjustment retry mechanism
    - Apply consistent implementation across all generator classes
    - Improve error logging for API error debugging
2025-04-07 10:12:38 +08:00
omahs
ca01a9e73f
Fix typos (#60)
* fix typo

* fix typo

* fix typos

* fix typos

* fix typos

* fix typo

* fix typos

* fix typos

* fix typos
2025-03-27 10:18:39 +08:00
justcrab
e778ebf82f
feat(logging): separate training log (#83)
* desparate part of traing logs v1

* fix train.py log

* optimize logging manage && fix monitor log

* feat(logging): desparate part of training logs v2

* delete no use code

* merge conflic

* delete chinese

---------

Co-authored-by: Ye Xiangle <yexiangle@mail.mindverse.ai>
Co-authored-by: Crabboss Mr <crabbossmr@CrabbossdeMacBook-Air.local>
2025-03-27 10:12:11 +08:00
Kevin
7f7d64210e Initial commit 2025-03-20 00:37:54 +08:00