Parcourir la source

Fix readme code example, require Python < 3.11 until supported (#374)

* Fix readme code example

* Require Python < 3.11 until it's supported
Alexander Borzunov il y a 2 ans
Parent
commit
895327a0ae
2 fichiers modifiés avec 2 ajouts et 2 suppressions
  1. 1 1
      README.md
  2. 1 1
      setup.cfg

+ 1 - 1
README.md

@@ -12,7 +12,7 @@ from transformers import AutoTokenizer
 from petals import AutoDistributedModelForCausalLM
 
 model_name = "enoch/llama-65b-hf"  # You can also use "bigscience/bloom" or "bigscience/bloomz"
-tokenizer = AutoTokenizer(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoDistributedModelForCausalLM.from_pretrained(model_name)
 # Embeddings & prompts are on your device, transformer blocks are distributed across the Internet
 

+ 1 - 1
setup.cfg

@@ -29,7 +29,7 @@ classifiers =
 package_dir =
     = src
 packages = find:
-python_requires = >=3.8
+python_requires = >=3.8,<3.11
 install_requires =
     torch>=1.12
     bitsandbytes==0.40.1.post1