Commit Graph

80 Commits

Author SHA1 Message Date
Kangle Deng 37de887d4e Code for colored voxels 2025-05-08 06:02:43 +00:00
Kangle Deng 855099347d Add examples 2025-05-07 17:13:04 +00:00
Kangle Deng 352585ce1f Add README 2025-05-07 05:58:23 +00:00
Kangle Deng 0af52df586 Add texture code 2025-05-07 05:46:16 +00:00
apun 51352e16f9 Update fine-tuning instructions in README 2025-04-30 01:19:41 -04:00
Ava Alison Pun 3a7679bf00 Merge branch 'main' of github.com:AvaLovelace1/LegoGPT 2025-04-30 01:17:07 -04:00
Ava Alison Pun 86a1beabfd Update finetuning script 2025-04-30 01:16:57 -04:00
apun 2757f6946a Update fine-tuning instructions in README 2025-04-30 01:09:24 -04:00
apun 3ac205aa88 Add prepare_finetuning_dataset to uv scripts 2025-04-30 01:09:11 -04:00
apun f11c02315d Remove loading parameters in LLM 2025-04-30 01:08:44 -04:00
apun e507767dfe Add prepare_finetuning_dataset script 2025-04-30 01:06:12 -04:00
apun 9fa2b6fe76 Update few-shot examples to use "lego" instead of "messages" 2025-04-29 23:13:15 -04:00
apun 51b3f97594 Update README 2025-04-29 22:43:26 -04:00
apun a583e3b52a Update infer script to output LEGO render 2025-04-29 22:32:14 -04:00
apun 00bd4303e4 Add return_dict option to LLM 2025-04-29 21:48:57 -04:00
apun 058be3fb39 remove print statement 2025-04-22 19:58:13 -04:00
apun 771c066466 output newline at end of sequence again 2025-04-22 19:56:55 -04:00
apun 5c97476fa5 Update installation instructions in README 2025-04-18 14:05:53 -04:00
Ava Alison Pun 9b21de1387 Merge branch 'main' of github.com:AvaLovelace1/LegoGPT 2025-04-18 13:04:18 -04:00
Ava Alison Pun 84caa0d620 Update finetuning script 2025-04-18 13:04:11 -04:00
apun 31458ba980 Add collision check to is_stable 2025-04-17 10:20:47 -04:00
apun 61374d85af Update logit masking & temperature defaults 2025-04-17 10:15:07 -04:00
apun fef38bcf3d Update README 2025-04-16 10:27:47 -04:00
apun 21ab24474f Add temperature increase options to LegoGPTConfig 2025-04-15 10:55:14 -04:00
apun 8d3c3761cd Update default temperature 2025-04-15 10:35:49 -04:00
apun 11e0f1516c Rename inference_masking to logit_masking 2025-04-15 10:35:33 -04:00
apun 2e7d04fc16 Revert "Revert "Add temperature increase after each rejection"" 2025-04-15 10:34:54 -04:00
apun c51fe33504 Add num_return_sequences 2025-04-13 11:14:37 -04:00
apun a9d5aabd8a Add out-of-bounds check to stability calculation 2025-04-11 21:47:36 -04:00
apun e6d962f2ad Add few-shot examples file 2025-04-11 13:47:39 -04:00
apun 57f648e02c Add few-shot instruction 2025-04-11 13:45:52 -04:00
apun 80e40accb3 Only print warning messages if max tries > 0 2025-04-11 11:21:49 -04:00
apun 76ac76d2b8 Only print warning messages if max tries > 0 2025-04-11 11:19:18 -04:00
apun 1c496acec2 Add zero-shot instruction 2025-04-11 10:28:49 -04:00
apun 4248258484 Separate "brick in library" check from "ill-formatted" check 2025-04-10 14:23:46 -04:00
apun aa838afe99 Add partial structure test 2025-04-10 13:59:40 -04:00
apun adb6e14b20 Update warning messages 2025-04-10 09:23:13 -04:00
apun f6a4142929 Rename use_inference_masking to use_logit_masking 2025-04-09 12:19:33 -04:00
apun 58a051a9ec Output number of bricks and regenerations in infer.py 2025-04-09 12:17:40 -04:00
apun 132ab420ed Enhance max_brick_rejections warning message with last generated brick and Lego structure details 2025-04-09 11:53:57 -04:00
apun ee98f304c9 Add top_p option to LegoGPT 2025-04-09 11:45:10 -04:00
apun 202309fe8d Increase default maximum brick rejections 2025-04-09 10:17:35 -04:00
apun 564932a6c2 Update .gitignore to include Python-generated files and virtual environments 2025-04-09 10:17:11 -04:00
apun a857b3668e Add imports to __init__ files for improved module structure 2025-04-09 10:17:02 -04:00
apun 83ceed17a9 Revert "Add temperature increase after each rejection"
This reverts commit 98bb1cd02f.
2025-04-09 10:06:02 -04:00
apun 98bb1cd02f Add temperature increase after each rejection 2025-04-08 17:52:48 -04:00
apun b546ca3751 Change final brick detection method to check for newline 2025-04-08 17:16:24 -04:00
apun dcc42ef52e Max new tokens number fix 2025-04-08 17:08:56 -04:00
apun 018646898c Update uv.lock 2025-04-08 17:02:29 -04:00
apun 0c30200266 Add top_k parameter for token sampling in LLM and LegoGPT 2025-04-08 17:02:23 -04:00