langchain/tests/integration_tests
Ravindra Marella b3988621c5
Add C Transformers for GGML Models (#5218)
# Add C Transformers for GGML Models
I created Python bindings for the GGML models:
https://github.com/marella/ctransformers

Currently it supports GPT-2, GPT-J, GPT-NeoX, LLaMA, MPT, etc. See
[Supported
Models](https://github.com/marella/ctransformers#supported-models).


It provides a unified interface for all models:

```python
from langchain.llms import CTransformers

llm = CTransformers(model='/path/to/ggml-gpt-2.bin', model_type='gpt2')

print(llm('AI is going to'))
```

It can be used with models hosted on the Hugging Face Hub:

```py
llm = CTransformers(model='marella/gpt-2-ggml')
```

It supports streaming:

```py
from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler

llm = CTransformers(model='marella/gpt-2-ggml', callbacks=[StreamingStdOutCallbackHandler()])
```

Please see [README](https://github.com/marella/ctransformers#readme) for
more details.
---------

Co-authored-by: Dev 2049 <dev.dev2049@gmail.com>
2023-05-25 13:42:44 -07:00
..
agent power bi api wrapper integration tests & bug fix (#4983) 2023-05-19 11:25:52 -04:00
cache Optimize the initialization method of GPTCache (#4522) 2023-05-11 16:15:23 -07:00
callbacks add tracing v2 env var (#4465) 2023-05-10 11:08:29 -07:00
chains Harrison/neo4j (#5078) 2023-05-22 07:31:48 -07:00
chat_models Harrison/vertex (#5049) 2023-05-24 15:51:12 -07:00
client Add Delete Session Method (#5193) 2023-05-24 21:06:03 +00:00
document_loaders Add Joplin document loader (#5153) 2023-05-24 12:31:55 -07:00
embeddings Harrison/vertex (#5049) 2023-05-24 15:51:12 -07:00
examples fix(document_loaders/telegram): fix pandas calls + add tests (#4806) 2023-05-16 14:35:25 -07:00
llms Add C Transformers for GGML Models (#5218) 2023-05-25 13:42:44 -07:00
memory Cassandra support for chat history (#4378) (#4764) 2023-05-15 23:43:09 -07:00
prompts Cleanup integration test dir (#3308) 2023-04-21 09:44:09 -07:00
retrievers tfidf retriever (#5114) 2023-05-24 10:02:09 -07:00
utilities Fix graphql tool (#4984) 2023-05-19 15:27:50 -07:00
vectorstores Vectara (#5069) 2023-05-24 01:24:58 -07:00
__init__.py initial commit 2022-10-24 14:51:15 -07:00
.env.example power bi api wrapper integration tests & bug fix (#4983) 2023-05-19 11:25:52 -04:00
conftest.py feat: improve pinecone tests (#2806) 2023-04-13 21:49:31 -07:00
test_document_transformers.py Contextual compression retriever (#2915) 2023-04-20 17:01:14 -07:00
test_nlp_text_splitters.py OptimizedPrompt -- k-shot example choice backed by semantic search (#91) 2022-11-09 21:15:42 -08:00
test_pdf_pagesplitter.py cleanup: unify 3 different pdf loaders, rename PagedPDFSplitter (#1615) 2023-03-13 23:06:50 -07:00
test_schema.py Add 'get_token_ids' method (#4784) 2023-05-22 13:17:26 +00:00
test_text_splitter.py Fix TextSplitter.from_tiktoken(#4361) 2023-05-08 16:36:38 -07:00