Travis Hammond 498c50348c Add DDPG and fix Gym wrapper (#1207) ...

* Fix Gym wrapper
- It was returning things in the wrong order
- Gym now differentiates between terminated and truncated

* Add DDPG

* Apply fixes

* Remove Result annotations

* Also remove Vec annotation

* rustfmt

* Various small improvements (avoid cloning, mutability, get clippy to pass, ...)

---------

Co-authored-by: Travis Hammond <travis.hammond@alexanderthamm.com>
Co-authored-by: Laurent <laurent.mazare@gmail.com>

2023-10-28 19:53:34 +01:00

..

bert

Handle LongStorage in pytorch checkpoints. (#1152)

2023-10-22 18:34:36 +01:00

bigcode

Adapt more examples to the updated safetensor api. (#947)

2023-09-23 21:26:03 +01:00

blip

Add a quantized blip model. (#1155)

2023-10-22 20:33:25 +01:00

convmixer

Convmixer example (#1074)

2023-10-11 19:51:10 +01:00

custom-ops

Remove some dead-code annotations. (#629)

2023-08-27 18:52:33 +01:00

dinov2

Adapt more examples to the updated safetensor api. (#947)

2023-09-23 21:26:03 +01:00

efficientnet

Adapt more examples to the updated safetensor api. (#947)

2023-09-23 21:26:03 +01:00

falcon

Adapt more examples to the updated safetensor api. (#947)

2023-09-23 21:26:03 +01:00

jina-bert

Use the hub model file when possible. (#1190)

2023-10-26 20:00:50 +01:00

llama

Adapt more examples to the updated safetensor api. (#947)

2023-09-23 21:26:03 +01:00

llama2-c

Infer the config for llama2-c. (#1208)

2023-10-28 19:00:39 +01:00

llama_multiprocess

Self-contained safetensors for the multiprocess llama example. (#950)

2023-09-24 06:54:49 +01:00

mistral

Quantized version of mistral. (#1009)

2023-09-30 18:25:47 +01:00

mnist-training

Add the optimizer trait. (#702)

2023-09-01 12:55:39 +01:00

musicgen

Use candle_nn::LSTM in encodec. (#1051)

2023-10-07 19:43:06 +01:00

phi

Add support for the phi-hermes finetuned model. (#1192)

2023-10-27 05:57:08 +01:00

quantized

Add support for Zephyr-7b in the quantized model. (#1124)

2023-10-18 17:31:26 +01:00

quantized-t5

Do not use the kv-cache on external key-value states. (#1054)

2023-10-07 22:37:19 +01:00

reinforcement-learning

Add DDPG and fix Gym wrapper (#1207)

2023-10-28 19:53:34 +01:00

replit-code

Add the quantized mpt model. (#1123)

2023-10-18 16:29:38 +01:00

resnet

Expose the larger resnets (50/101/152) in the example. (#1131)

2023-10-19 13:48:28 +01:00

segment-anything

Add negative prompts to segment-anything. (#1000)

2023-09-30 06:17:42 +01:00

stable-diffusion

Mention the flash-attention restriction in the readme. (#1158)

2023-10-23 10:26:56 +01:00

stable-lm

Quantized version of StableLM. (#1058)

2023-10-08 15:42:38 +01:00

t5

Add some reinforcement learning example. (#1090)

2023-10-14 16:46:43 +01:00

vit

Readme updates. (#1134)

2023-10-20 09:08:39 +01:00

whisper

Simd128 optimized q8k vecdot. (#1026)

2023-10-03 15:29:48 +01:00

wuerstchen

Remove some unusued bits. (#1067)

2023-10-09 19:49:57 +01:00

yolo-v3

Make func cloneable. (#1137)

2023-10-20 16:28:50 +01:00

yolo-v8

Add fuse-conv-bn method for Conv2d (#1196)

2023-10-27 15:56:50 +01:00