├── .github └── workflows │ ├── build_and_release.yaml │ ├── test_dev.yaml │ └── test_main.yaml ├── .gitignore ├── AGENTS.md ├── Cargo.lock ├── Cargo.toml ├── README.md ├── examples ├── embedding.rs ├── fill_mask.rs ├── reranker │ ├── corpus │ │ ├── machine_learning_doc.txt │ │ ├── math_doc.txt │ │ ├── misc_doc.txt │ │ └── physics_doc.txt │ └── main.rs ├── sentiment_analysis.rs ├── text_generation.rs ├── text_generation_streaming.rs ├── text_generation_tools.rs ├── xml_parser.rs ├── xml_parser_streaming.rs └── zero_shot.rs ├── license.md ├── src ├── core │ ├── cache.rs │ ├── config.rs │ ├── error.rs │ ├── message.rs │ └── mod.rs ├── lib.rs ├── loaders.rs ├── models │ ├── components │ │ ├── attention.rs │ │ ├── layers.rs │ │ ├── mod.rs │ │ └── quantization.rs │ ├── generation │ │ ├── logits.rs │ │ ├── mod.rs │ │ ├── params.rs │ │ └── sampling.rs │ ├── implementations │ │ ├── gemma3.rs │ │ ├── mod.rs │ │ ├── modernbert.rs │ │ ├── qwen3.rs │ │ ├── qwen3_embeddings.rs │ │ └── qwen3_reranker.rs │ └── mod.rs └── pipelines │ ├── base.rs │ ├── embedding_pipeline │ ├── builder.rs │ ├── mod.rs │ ├── model.rs │ └── pipeline.rs │ ├── fill_mask_pipeline │ ├── builder.rs │ ├── mod.rs │ ├── model.rs │ └── pipeline.rs │ ├── mod.rs │ ├── reranker_pipeline │ ├── builder.rs │ ├── mod.rs │ ├── model.rs │ └── pipeline.rs │ ├── sentiment_analysis_pipeline │ ├── builder.rs │ ├── mod.rs │ ├── model.rs │ └── pipeline.rs │ ├── text_generation_pipeline │ ├── base_pipeline.rs │ ├── builder.rs │ ├── mod.rs │ ├── model.rs │ ├── parser.rs │ ├── pipeline.rs │ ├── streaming │ │ ├── completion_stream.rs │ │ ├── event_stream.rs │ │ └── mod.rs │ ├── tools.rs │ └── xml_pipeline.rs │ ├── utils │ ├── builder.rs │ └── mod.rs │ └── zero_shot_classification_pipeline │ ├── builder.rs │ ├── mod.rs │ ├── model.rs │ └── pipeline.rs ├── tests ├── common │ └── mod.rs ├── embedding_pipeline_tests │ ├── basic_embedding.rs │ ├── batch_embedding.rs │ └── main.rs ├── fill_mask_pipeline_tests │ ├── basic_fill_mask.rs │ └── main.rs ├── misc_pipeline_tests │ ├── main.rs │ ├── multi_pipeline.rs │ ├── tool_calling.rs │ ├── tool_error_handling.rs │ └── tool_management.rs ├── sentiment_analysis_pipeline_tests │ ├── basic_sentiment_analysis.rs │ └── main.rs ├── text_generation_pipeline_tests │ ├── basic_text_generation.rs │ └── main.rs └── zero_shot_pipeline_tests │ ├── basic_zero_shot_classification.rs │ └── main.rs └── tool_macro ├── Cargo.toml └── src └── lib.rs /.github/workflows/build_and_release.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/.github/workflows/build_and_release.yaml -------------------------------------------------------------------------------- /.github/workflows/test_dev.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/.github/workflows/test_dev.yaml -------------------------------------------------------------------------------- /.github/workflows/test_main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/.github/workflows/test_main.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/.gitignore -------------------------------------------------------------------------------- /AGENTS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/AGENTS.md -------------------------------------------------------------------------------- /Cargo.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/Cargo.lock -------------------------------------------------------------------------------- /Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/Cargo.toml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/README.md -------------------------------------------------------------------------------- /examples/embedding.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/embedding.rs -------------------------------------------------------------------------------- /examples/fill_mask.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/fill_mask.rs -------------------------------------------------------------------------------- /examples/reranker/corpus/machine_learning_doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/reranker/corpus/machine_learning_doc.txt -------------------------------------------------------------------------------- /examples/reranker/corpus/math_doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/reranker/corpus/math_doc.txt -------------------------------------------------------------------------------- /examples/reranker/corpus/misc_doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/reranker/corpus/misc_doc.txt -------------------------------------------------------------------------------- /examples/reranker/corpus/physics_doc.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/reranker/corpus/physics_doc.txt -------------------------------------------------------------------------------- /examples/reranker/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/reranker/main.rs -------------------------------------------------------------------------------- /examples/sentiment_analysis.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/sentiment_analysis.rs -------------------------------------------------------------------------------- /examples/text_generation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/text_generation.rs -------------------------------------------------------------------------------- /examples/text_generation_streaming.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/text_generation_streaming.rs -------------------------------------------------------------------------------- /examples/text_generation_tools.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/text_generation_tools.rs -------------------------------------------------------------------------------- /examples/xml_parser.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/xml_parser.rs -------------------------------------------------------------------------------- /examples/xml_parser_streaming.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/xml_parser_streaming.rs -------------------------------------------------------------------------------- /examples/zero_shot.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/examples/zero_shot.rs -------------------------------------------------------------------------------- /license.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/license.md -------------------------------------------------------------------------------- /src/core/cache.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/core/cache.rs -------------------------------------------------------------------------------- /src/core/config.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/core/config.rs -------------------------------------------------------------------------------- /src/core/error.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/core/error.rs -------------------------------------------------------------------------------- /src/core/message.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/core/message.rs -------------------------------------------------------------------------------- /src/core/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/core/mod.rs -------------------------------------------------------------------------------- /src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/lib.rs -------------------------------------------------------------------------------- /src/loaders.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/loaders.rs -------------------------------------------------------------------------------- /src/models/components/attention.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/components/attention.rs -------------------------------------------------------------------------------- /src/models/components/layers.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/components/layers.rs -------------------------------------------------------------------------------- /src/models/components/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/components/mod.rs -------------------------------------------------------------------------------- /src/models/components/quantization.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/components/quantization.rs -------------------------------------------------------------------------------- /src/models/generation/logits.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/generation/logits.rs -------------------------------------------------------------------------------- /src/models/generation/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/generation/mod.rs -------------------------------------------------------------------------------- /src/models/generation/params.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/generation/params.rs -------------------------------------------------------------------------------- /src/models/generation/sampling.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/generation/sampling.rs -------------------------------------------------------------------------------- /src/models/implementations/gemma3.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/implementations/gemma3.rs -------------------------------------------------------------------------------- /src/models/implementations/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/implementations/mod.rs -------------------------------------------------------------------------------- /src/models/implementations/modernbert.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/implementations/modernbert.rs -------------------------------------------------------------------------------- /src/models/implementations/qwen3.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/implementations/qwen3.rs -------------------------------------------------------------------------------- /src/models/implementations/qwen3_embeddings.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/implementations/qwen3_embeddings.rs -------------------------------------------------------------------------------- /src/models/implementations/qwen3_reranker.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/implementations/qwen3_reranker.rs -------------------------------------------------------------------------------- /src/models/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/models/mod.rs -------------------------------------------------------------------------------- /src/pipelines/base.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/base.rs -------------------------------------------------------------------------------- /src/pipelines/embedding_pipeline/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/embedding_pipeline/builder.rs -------------------------------------------------------------------------------- /src/pipelines/embedding_pipeline/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/embedding_pipeline/mod.rs -------------------------------------------------------------------------------- /src/pipelines/embedding_pipeline/model.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/embedding_pipeline/model.rs -------------------------------------------------------------------------------- /src/pipelines/embedding_pipeline/pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/embedding_pipeline/pipeline.rs -------------------------------------------------------------------------------- /src/pipelines/fill_mask_pipeline/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/fill_mask_pipeline/builder.rs -------------------------------------------------------------------------------- /src/pipelines/fill_mask_pipeline/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/fill_mask_pipeline/mod.rs -------------------------------------------------------------------------------- /src/pipelines/fill_mask_pipeline/model.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/fill_mask_pipeline/model.rs -------------------------------------------------------------------------------- /src/pipelines/fill_mask_pipeline/pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/fill_mask_pipeline/pipeline.rs -------------------------------------------------------------------------------- /src/pipelines/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/mod.rs -------------------------------------------------------------------------------- /src/pipelines/reranker_pipeline/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/reranker_pipeline/builder.rs -------------------------------------------------------------------------------- /src/pipelines/reranker_pipeline/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/reranker_pipeline/mod.rs -------------------------------------------------------------------------------- /src/pipelines/reranker_pipeline/model.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/reranker_pipeline/model.rs -------------------------------------------------------------------------------- /src/pipelines/reranker_pipeline/pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/reranker_pipeline/pipeline.rs -------------------------------------------------------------------------------- /src/pipelines/sentiment_analysis_pipeline/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/sentiment_analysis_pipeline/builder.rs -------------------------------------------------------------------------------- /src/pipelines/sentiment_analysis_pipeline/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/sentiment_analysis_pipeline/mod.rs -------------------------------------------------------------------------------- /src/pipelines/sentiment_analysis_pipeline/model.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/sentiment_analysis_pipeline/model.rs -------------------------------------------------------------------------------- /src/pipelines/sentiment_analysis_pipeline/pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/sentiment_analysis_pipeline/pipeline.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/base_pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/base_pipeline.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/builder.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/mod.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/model.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/model.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/parser.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/parser.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/pipeline.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/streaming/completion_stream.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/streaming/completion_stream.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/streaming/event_stream.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/streaming/event_stream.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/streaming/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/streaming/mod.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/tools.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/tools.rs -------------------------------------------------------------------------------- /src/pipelines/text_generation_pipeline/xml_pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/text_generation_pipeline/xml_pipeline.rs -------------------------------------------------------------------------------- /src/pipelines/utils/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/utils/builder.rs -------------------------------------------------------------------------------- /src/pipelines/utils/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/utils/mod.rs -------------------------------------------------------------------------------- /src/pipelines/zero_shot_classification_pipeline/builder.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/zero_shot_classification_pipeline/builder.rs -------------------------------------------------------------------------------- /src/pipelines/zero_shot_classification_pipeline/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/zero_shot_classification_pipeline/mod.rs -------------------------------------------------------------------------------- /src/pipelines/zero_shot_classification_pipeline/model.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/zero_shot_classification_pipeline/model.rs -------------------------------------------------------------------------------- /src/pipelines/zero_shot_classification_pipeline/pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/src/pipelines/zero_shot_classification_pipeline/pipeline.rs -------------------------------------------------------------------------------- /tests/common/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/common/mod.rs -------------------------------------------------------------------------------- /tests/embedding_pipeline_tests/basic_embedding.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/embedding_pipeline_tests/basic_embedding.rs -------------------------------------------------------------------------------- /tests/embedding_pipeline_tests/batch_embedding.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/embedding_pipeline_tests/batch_embedding.rs -------------------------------------------------------------------------------- /tests/embedding_pipeline_tests/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/embedding_pipeline_tests/main.rs -------------------------------------------------------------------------------- /tests/fill_mask_pipeline_tests/basic_fill_mask.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/fill_mask_pipeline_tests/basic_fill_mask.rs -------------------------------------------------------------------------------- /tests/fill_mask_pipeline_tests/main.rs: -------------------------------------------------------------------------------- 1 | mod basic_fill_mask; 2 | -------------------------------------------------------------------------------- /tests/misc_pipeline_tests/main.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/misc_pipeline_tests/main.rs -------------------------------------------------------------------------------- /tests/misc_pipeline_tests/multi_pipeline.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/misc_pipeline_tests/multi_pipeline.rs -------------------------------------------------------------------------------- /tests/misc_pipeline_tests/tool_calling.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/misc_pipeline_tests/tool_calling.rs -------------------------------------------------------------------------------- /tests/misc_pipeline_tests/tool_error_handling.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/misc_pipeline_tests/tool_error_handling.rs -------------------------------------------------------------------------------- /tests/misc_pipeline_tests/tool_management.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/misc_pipeline_tests/tool_management.rs -------------------------------------------------------------------------------- /tests/sentiment_analysis_pipeline_tests/basic_sentiment_analysis.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/sentiment_analysis_pipeline_tests/basic_sentiment_analysis.rs -------------------------------------------------------------------------------- /tests/sentiment_analysis_pipeline_tests/main.rs: -------------------------------------------------------------------------------- 1 | mod basic_sentiment_analysis; 2 | -------------------------------------------------------------------------------- /tests/text_generation_pipeline_tests/basic_text_generation.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/text_generation_pipeline_tests/basic_text_generation.rs -------------------------------------------------------------------------------- /tests/text_generation_pipeline_tests/main.rs: -------------------------------------------------------------------------------- 1 | mod basic_text_generation; 2 | -------------------------------------------------------------------------------- /tests/zero_shot_pipeline_tests/basic_zero_shot_classification.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tests/zero_shot_pipeline_tests/basic_zero_shot_classification.rs -------------------------------------------------------------------------------- /tests/zero_shot_pipeline_tests/main.rs: -------------------------------------------------------------------------------- 1 | mod basic_zero_shot_classification; 2 | -------------------------------------------------------------------------------- /tool_macro/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tool_macro/Cargo.toml -------------------------------------------------------------------------------- /tool_macro/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ljt019/transformers/HEAD/tool_macro/src/lib.rs --------------------------------------------------------------------------------