Commit Graph

83 Commits

Author SHA1 Message Date
Ava Pun 226bb0673f Update README.md 2025-05-05 18:50:53 -07:00
apun 27afa8dd38 Add output image 2025-05-05 21:46:51 -04:00
apun 7234d0421e Add example inference interaction 2025-05-05 21:46:25 -04:00
apun 680bd35b64 Add path to ImportLDraw module in render script 2025-05-05 21:33:12 -04:00
apun 350fdb265d Add LDraw library download instructions to README 2025-05-05 21:27:05 -04:00
apun ca5937842b Add render script to pyproject.toml 2025-05-05 19:01:53 -04:00
apun 197c4692a8 output folder fixes 2025-05-01 04:27:02 -04: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