 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 |
|
|
apun
|
bb0acfa239
|
Add inference masking option for LEGO brick generation
|
2025-04-08 17:01:36 -04:00 |
|
|
apun
|
6aaa64b13d
|
Add warnings when rejection/regeneration reaches max # of tries
|
2025-04-08 15:58:25 -04:00 |
|
|
apun
|
ac48cfc6e7
|
Add error handling for stability score computation in LegoStructure
|
2025-04-08 15:33:15 -04:00 |
|
|
Ava Alison Pun
|
fe265c6123
|
Update finetuning script
|
2025-04-08 14:43:20 -04:00 |
|
|
apun
|
d2ed5cbcb1
|
Fix stability check in legogpt.py to call is_stable() method
|
2025-04-08 09:19:31 -04:00 |
|
|
apun
|
ce27126502
|
Update llm.py to set model torch_dtype to 'auto'
|
2025-04-08 09:19:14 -04:00 |
|
|
apun
|
d3e203930c
|
LegoStructure to_ldr fix
|
2025-04-04 12:02:11 -04:00 |
|
|
apun
|
bc1b68f41f
|
Add to_ldr function to LegoStructure
|
2025-04-04 09:28:07 -04:00 |
|
|
apun
|
45915e8783
|
Change python version
|
2025-04-03 21:31:50 -04:00 |
|
|
apun
|
2f15434fc9
|
Create infer uv script
|
2025-04-01 09:48:04 -04:00 |
|
|
apun
|
6a24c14466
|
Refactor LegoGPT initialization to use config dataclass
|
2025-03-31 23:01:13 -04:00 |
|
|
apun
|
15d508f86a
|
Move LEGO generation parameters for improved configurability
|
2025-03-31 16:22:39 -04:00 |
|
|
apun
|
f615a37e0a
|
Add physics-based rollback
|
2025-03-31 16:17:12 -04:00 |
|
|
apun
|
a5db015d0b
|
Add LegoGPT ability to start from partial LEGO
|
2025-03-31 13:13:39 -04:00 |
|
|
apun
|
c608390d65
|
Inference masking fix (allow ")<eot>" in addition to ")\n<eot>")
|
2025-03-31 10:49:18 -04:00 |
|
|
apun
|
72ebc032e5
|
Remove flash attention since it is already implemented in BetterTransformer
|
2025-03-31 10:40:32 -04:00 |
|
|
apun
|
fa09dc37b5
|
Add test_finetuned_llm test
|
2025-03-31 10:39:59 -04:00 |
|
|
apun
|
4b18b54787
|
Update LegoStructure
|
2025-03-24 15:36:47 -04:00 |
|
|
apun
|
d2f075ec3e
|
Merge
|
2025-03-23 11:44:07 -04:00 |
|
|
apun
|
9b8918081b
|
Move finetuning script
|
2025-03-23 11:41:02 -04:00 |
|
|
apun
|
ac3c23d2ad
|
Add finetuning script
|
2025-03-23 11:09:57 -04:00 |
|
|
apun
|
997e75035a
|
Add max_bricks parameter to limit the number of generated bricks in LegoGPT
|
2025-03-22 21:19:57 -04:00 |
|
|
apun
|
474bcc1bd6
|
Add option to return rejection reasons in LegoGPT inference
|
2025-03-22 21:15:09 -04:00 |
|
|
apun
|
0c61639be3
|
Add out-of-bounds check to rejection sampling
|
2025-03-22 21:14:43 -04:00 |
|
|
apun
|
04888f5dd1
|
Add collision check to rejection sampling
|
2025-03-22 21:01:04 -04:00 |
|
|
apun
|
8247549dd0
|
Add collision check to rejection sampling
|
2025-03-22 20:52:15 -04:00 |
|
|
apun
|
103f5397f2
|
Add stability analysis to LegoStructure
|
2025-03-22 20:51:49 -04:00 |
|
|
apun
|
6c35b0194f
|
Add floating brick detection to LegoStructure
|
2025-03-22 20:21:41 -04:00 |
|
|
apun
|
1a1d13255d
|
Add collision detection and ground level check to LegoStructure
|
2025-03-22 20:16:10 -04:00 |
|
|
apun
|
04c050c228
|
First commit
|
2025-03-22 17:21:42 -04:00 |
|