github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-03 14:18:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(petals) allows to run models that aren't Bloom (Support for LLama and newer models) (#8356)

Browse Source 
In this PR:

- Removed restricted model loading logic for Petals-Bloom
- Removed petals imports (DistributedBloomForCausalLM,
BloomTokenizerFast)
- Instead imported more generalized versions of loader
(AutoDistributedModelForCausalLM, AutoTokenizer)
- Updated the Petals example notebook to allow for a successful
installation of Petals in Apple Silicon Macs

- Tag maintainer: @hwchase17, @baskaryan

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
Karan V 2023-07-28 06:31:04 +05:30 committed by GitHub
parent e758e9e7f5
commit a003a0baf6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/extras/integrations/llms/petals_example.ipynb
View File

@ -16,7 +16,9 @@
"metadata": {}, "metadata": {},
"source": [ "source": [
"## Install petals\n", "## Install petals\n",
"The `petals` package is required to use the Petals API. Install `petals` using `pip3 install petals`." "The `petals` package is required to use the Petals API. Install `petals` using `pip3 install petals`.\n",
"\n",
"For Apple Silicon(M1/M2) users please follow this guide [https://github.com/bigscience-workshop/petals/issues/147#issuecomment-1365379642](https://github.com/bigscience-workshop/petals/issues/147#issuecomment-1365379642) to install petals "
] ]
}, },
{ {
@ -62,7 +64,7 @@
}, },
"outputs": [ "outputs": [
{ {
"name": "stdin", "name": "stdout",
"output_type": "stream", "output_type": "stream",
"text": [ "text": [
" ········\n" " ········\n"

10
libs/langchain/langchain/llms/petals.py
View File

@ -93,12 +93,14 @@ class Petals(LLM):
values, "huggingface_api_key", "HUGGINGFACE_API_KEY" values, "huggingface_api_key", "HUGGINGFACE_API_KEY"
) )
try: try:
from petals import DistributedBloomForCausalLM from petals import AutoDistributedModelForCausalLM
from transformers import BloomTokenizerFast from transformers import AutoTokenizer
model_name = values["model_name"] model_name = values["model_name"]
values["tokenizer"] = BloomTokenizerFast.from_pretrained(model_name) values["tokenizer"] = AutoTokenizer.from_pretrained(model_name)
values["client"] = DistributedBloomForCausalLM.from_pretrained(model_name) values["client"] = AutoDistributedModelForCausalLM.from_pretrained(
model_name
)
values["huggingface_api_key"] = huggingface_api_key values["huggingface_api_key"] = huggingface_api_key
except ImportError: except ImportError: