Update block.py

pull/277/head
justheuristic 1 year ago committed by GitHub
parent 5819c1a84b
commit 8bb8e86a6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -11,7 +11,7 @@ import transformers
from transformers.models.bloom.modeling_bloom import BloomBlock, _expand_mask, _make_causal_mask, build_alibi_tensor
if not os.getenv("PETALS_IGNORE_DEPENDENCY_VERSION"):
assert transformers.__version__.startswith("4.25."), "Please install transformers 4.25.1"
assert transformers.__version__.startswith("4.26."), "Please install transformers 4.26.1"
class WrappedBloomBlock(BloomBlock):

Loading…
Cancel
Save