Spaces:

yahma
/

BLOOMZ_Compare

Build error

gururise commited on Jan 31, 2023

Commit

a100dd2

1 Parent(s): 1b9f1cc

update model cache location

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,8 @@ import codecs
 from datetime import datetime
 import os
-os.environ['TRANSFORMERS_CACHE'] = '/data/.modelcache/huggingface/hub/'
-os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:516"
 from transformers import BloomTokenizerFast
 from petals.client import DistributedBloomForCausalLM

 from datetime import datetime
 import os
+#os.environ['TRANSFORMERS_CACHE'] = '/data/.modelcache/huggingface/hub/'
+#os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:516"
 from transformers import BloomTokenizerFast
 from petals.client import DistributedBloomForCausalLM