 Brett Kuprel
|
1ffdef9a56
|
clamp in place
|
2022-07-10 08:07:54 -04:00 |
|
|
Brett Kuprel
|
dba3f11b3f
|
faster inference with cuda/cudnn backends flags
|
2022-07-09 06:48:51 -04:00 |
|
|
Brett Kuprel
|
9eb5633931
|
clamp tokens to appropriate bounds
|
2022-07-08 09:19:28 -04:00 |
|
|
Brett Kuprel
|
736904ef2f
|
fix typing
|
2022-07-07 17:18:30 -04:00 |
|
|
Brett Kuprel
|
da62298f06
|
support bfloat16
|
2022-07-07 08:21:20 -04:00 |
|
|
Brett Kuprel
|
1eb56737d8
|
added grid_size parameter to generate a grid of images
|
2022-07-02 08:45:49 -04:00 |
|
|
Brett Kuprel
|
df9aa6f915
|
sort -> topk, prev_token_and_index -> prev_token, token_index
|
2022-06-30 09:04:11 -04:00 |
|
|
Brett Kuprel
|
17c96fe110
|
works with cuda
|
2022-06-28 21:28:36 -04:00 |
|
|
Brett Kuprel
|
aef24ea157
|
torch.no_grad(), cleanup
|
2022-06-28 12:16:44 -04:00 |
|
|
Brett Kuprel
|
18e6a9852f
|
license and cleanup
|
2022-06-27 14:34:10 -04:00 |
|
|
Brett Kuprel
|
018414a5c3
|
fixed relative imports
|
2022-06-27 12:43:47 -04:00 |
|