langchain/tests/integration_tests/llms
Alex Rad bd780a8223
Add support for rwkv (#2422)
This adds support for running RWKV with pytorch. 

https://github.com/hwchase17/langchain/issues/2398

This does not yet support  rwkv.cpp
2023-04-06 14:41:06 -07:00
..
__init__.py initial commit 2022-10-24 14:51:15 -07:00
test_ai21.py Harrison/cohere experimental (#638) 2023-01-17 22:28:55 -08:00
test_aleph_alpha.py add aleph alpha llm (#1207) 2023-02-22 10:37:36 -08:00
test_anthropic.py add async support for anthropic (#2114) 2023-03-28 22:49:14 -04:00
test_banana.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
test_cerebrium.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_cohere.py Harrison/llm saving (#331) 2022-12-13 06:46:01 -08:00
test_forefrontai.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_gooseai.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_gpt4all.py Harrison/gpt4all (#2366) 2023-04-04 06:49:17 -07:00
test_huggingface_endpoint.py Harrison/inference endpoint (#861) 2023-02-06 18:14:25 -08:00
test_huggingface_hub.py Harrison/llm saving (#331) 2022-12-13 06:46:01 -08:00
test_huggingface_pipeline.py Harrison/version 0040 (#366) 2022-12-17 07:53:22 -08:00
test_llamacpp.py Harrison/llama (#2314) 2023-04-02 14:57:45 -07:00
test_manifest.py Harrison/fix lint (#138) 2022-11-14 08:55:59 -08:00
test_modal.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
test_nlpcloud.py Harrison/llm saving (#331) 2022-12-13 06:46:01 -08:00
test_openai.py (OpenAI) Add model_name to LLMResult.llm_output (#1713) 2023-03-16 21:55:55 -07:00
test_petals.py Add GooseAI, CerebriumAI, Petals, ForefrontAI (#981) 2023-02-13 21:20:19 -08:00
test_promptlayer_openai.py Harrison/llm integrations (#1039) 2023-02-13 22:06:25 -08:00
test_propmptlayer_openai_chat.py Harrison/prompt layer (#1547) 2023-03-08 21:24:27 -08:00
test_replicate.py Add replicate take 2 (#2077) 2023-03-28 11:56:57 -07:00
test_rwkv.py Add support for rwkv (#2422) 2023-04-06 14:41:06 -07:00
test_self_hosted_llm.py Harrison/self hosted runhouse (#1154) 2023-02-19 09:53:45 -08:00
test_stochasticai.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
test_writer.py Add Writer, Banana, Modal, StochasticAI (#1270) 2023-02-24 06:58:58 -08:00
utils.py Harrison/improve cache (#368) 2022-12-18 16:22:42 -05:00