candle/candle-examples/examples
LongYinan cb02b389d5
Fix reinforcement learning example (#2837)
2025-03-26 16:27:45 +01:00
..
based Add Based LLM from Hazy Research. (#2411) 2024-08-12 21:21:19 +02:00
beit onnx: fix pad, unsqueeze (#2317) 2024-07-23 23:10:57 +02:00
bert bert attention mask (#1934) 2024-08-01 08:26:19 +02:00
bigcode Adapt more examples to the updated safetensor api. (#947) 2023-09-23 21:26:03 +01:00
blip onnx: fix pad, unsqueeze (#2317) 2024-07-23 23:10:57 +02:00
chatglm Add the custom tokenizer. (#1686) 2024-02-09 17:36:50 +01:00
chinese_clip feat: intergrate chinese clip and add example (#2555) 2024-10-10 15:18:55 +02:00
clip Add the SigLIP model. (#2515) 2024-09-28 23:48:00 +02:00
codegeex4-9b fix: fix the codegeex4 model examples and transformers model (#2738) 2025-01-25 17:41:12 +01:00
colpali Add ColPali (#2524) 2024-10-01 11:48:39 +02:00
convmixer Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
convnext Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
custom-ops Cuda acceleration for quantized model. (#1754) 2024-02-25 18:11:47 +01:00
debertav2 Adds DebertaV2/V3 (#2743) 2025-01-29 08:59:28 +01:00
deepseekv2 Implement DeepSeek V2 (#2744) 2025-02-19 10:51:01 +01:00
depth_anything_v2 make DepthAnythingV2 more reusable (#2675) 2024-12-21 12:06:03 +01:00
dinov2 Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
dinov2reg4 Add DINOv2Reg4 + PlantCLEF2024 (#2293) 2024-06-29 11:49:15 +02:00
distilbert Distibert (#1366) 2023-11-24 15:09:14 +00:00
efficientnet Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
efficientvit Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
encodec Make the RNN configs accessible from the models. (#2541) 2024-10-04 14:22:23 +02:00
eva2 onnx: fix pad, unsqueeze (#2317) 2024-07-23 23:10:57 +02:00
falcon Use the new hub helper function. (#1484) 2023-12-26 09:44:30 +01:00
fastvit FastViT fixes. (#2452) 2024-08-28 11:20:09 +02:00
flux Streamline the glm4 example. (#2694) 2024-12-31 09:21:41 +01:00
gemma Add Gemma 3 1b IT toe Gemma examples (#2809) 2025-03-16 17:00:48 +01:00
glm4 Fix GLM4 alignment issue (#2723) 2025-01-20 22:51:46 +01:00
granite Adding Granite 7b Instruct model example (#2487) 2024-09-21 11:52:01 +02:00
gte-qwen Support embedding model gte-Qwen1.5-7B-instruct (#2190) 2024-05-16 21:34:10 +02:00
helium Remove the old MFA gemm kernels. (#2742) 2025-01-26 20:36:31 +01:00
hiera Add Hiera vision model. (#2382) 2024-08-01 11:59:22 +02:00
jina-bert Fix cargo fmt. (#2383) 2024-08-01 14:19:41 +02:00
llama Add the SmolLM2 models. (#2595) 2024-11-03 17:11:12 +01:00
llama2-c Explicit caching in llama2.c. 2024-02-22 10:22:03 +01:00
llama_multiprocess Apply rustfmt. (#2421) 2024-08-16 18:57:14 +02:00
llava onnx: fix pad, unsqueeze (#2317) 2024-07-23 23:10:57 +02:00
mamba Support alternative dtypes for mamba (#2036) 2024-04-10 18:10:01 +02:00
mamba-minimal Lint fixes introduced with Rust 1.83 (#2646) 2024-11-28 23:00:21 +01:00
marian-mt Fix the marian tokenizer importer. (#2426) 2024-08-17 20:58:40 +02:00
metavoice upgrade half library to fix rand (#2806) 2025-03-14 09:01:54 +01:00
mimi Make the RNN configs accessible from the models. (#2541) 2024-10-04 14:22:23 +02:00
mistral Support for mistral-nemo. (#2396) 2024-08-04 19:52:40 +02:00
mixtral Use BF16 on metal when possible. (#2378) 2024-08-01 10:48:58 +02:00
mnist-training fixed rand import for mnist-training (#2833) 2025-03-26 08:10:03 +01:00
mobileclip Remove some extra whitelines. (#2513) 2024-09-28 14:41:28 +02:00
mobilenetv4 MobileCLIP models S1 and S2 (#2454) 2024-08-29 15:38:58 +02:00
mobileone Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
modernbert ModernBERT model (#2713) 2025-01-13 08:39:27 +01:00
moondream use moondream1 model/revision for moondream example (#2748) 2025-01-28 22:19:54 +01:00
musicgen Clippy fixes for 1.81.0. (#2461) 2024-09-05 23:46:55 +02:00
nvembed_v2 Add Nvembed v2 model (#2649) 2024-12-03 10:56:01 +01:00
olmo Add Olmo models (#2127) 2024-04-26 11:02:51 +02:00
onnx Update docs to reflect current usage of example (#1610) 2024-02-04 11:59:47 +01:00
paligemma Add PaliGemma. (#2519) 2024-09-29 19:56:56 +02:00
parler-tts Fix for parler-tts, do not add the last slice of padding tokens. (#2442) 2024-08-22 23:22:03 +02:00
phi phi-4-mini (#2790) 2025-03-01 10:07:29 +01:00
pixtral Pixtral polishing. (#2522) 2024-09-30 21:23:54 +02:00
quantized Add the SmolLM2 models. (#2595) 2024-11-03 17:11:12 +01:00
quantized-phi Fixes for running Phi-4 quantized. (#2714) 2025-01-13 14:35:33 +01:00
quantized-qwen2-instruct add quantized qwen2 (#2329) 2024-07-12 10:00:03 +02:00
quantized-t5 Move the tensor-tools binary in a separate crate. (#1969) 2024-03-30 15:49:37 +01:00
qwen Support for the new Qwen2 models. (#2257) 2024-06-07 10:51:50 +01:00
recurrent-gemma Add a quantized version of recurrent-gemma. (#2054) 2024-04-13 20:07:01 +02:00
reinforcement-learning Fix reinforcement learning example (#2837) 2025-03-26 16:27:45 +01:00
replit-code Quantized GGUF style (#1523) 2024-01-17 10:27:58 +01:00
repvgg Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
resnet Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
rwkv Fix the model path for rwkv. (#1825) 2024-03-09 11:21:48 +01:00
segformer Update the image crate + use the re-exported version. (#1893) 2024-03-21 10:56:41 +01:00
segment-anything onnx: fix pad, unsqueeze (#2317) 2024-07-23 23:10:57 +02:00
siglip Parse the json config for siglip models. (#2800) 2025-03-09 14:01:09 +01:00
silero-vad Clippy fixes for 1.81.0. (#2461) 2024-09-05 23:46:55 +02:00
splade Add BertForMaskedLM to support SPLADE Models (#2550) 2024-10-07 23:28:21 +02:00
stable-diffusion upgrade half library to fix rand (#2806) 2025-03-14 09:01:54 +01:00
stable-diffusion-3 Support Skip Layer Guidance (SLG) for Stable Diffusion 3.5 Medium (#2590) 2024-11-01 18:10:40 +01:00
stable-lm Avoid copying the data on squeeze and unsqueeze. (#1884) 2024-03-20 13:04:36 +01:00
starcoder2 Add the StarCoder2 model. (#1779) 2024-02-28 21:02:41 +01:00
stella-en-v5 Adds support for stella_en_v5 embedding model -400M variant (#2608) 2024-11-29 09:01:08 +01:00
t5 Expose the t5 config fields + allow t5-large. (#1987) 2024-04-01 20:58:34 +02:00
trocr onnx: fix pad, unsqueeze (#2317) 2024-07-23 23:10:57 +02:00
vgg Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
vit Move the image tensor to the appropriate device. (#1856) 2024-03-16 22:25:46 +01:00
whisper Fix for whisper example. rand::distribution is now rand::distr (#2811) 2025-03-16 19:14:55 +01:00
whisper-microphone fixed rand imports for whisper-microphone example (#2834) 2025-03-26 08:09:27 +01:00
wuerstchen Remove some unusued bits. (#1067) 2023-10-09 19:49:57 +01:00
xlm-roberta Added XLMRobertaModel for Reranking (#2686) 2024-12-30 11:16:57 +01:00
yi Fix token generation in bilingual models (non-English outputs) (#1668) 2024-02-06 12:03:53 +01:00
yolo-v3 Clippy fixes for 1.81.0. (#2461) 2024-09-05 23:46:55 +02:00
yolo-v8 Clippy fixes for 1.81.0. (#2461) 2024-09-05 23:46:55 +02:00
onnx_basics.rs [ONNX] Support a couple more ops. (#1284) 2023-11-06 22:44:58 +01:00