Spaces:
Runtime error
Runtime error
gmerrill
commited on
Commit
·
d108f97
1
Parent(s):
20165cd
update
Browse files
main.py
CHANGED
@@ -33,8 +33,6 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
|
33 |
result = subprocess.run('pwd && ls -lH && find /.cache/huggingface/hub && find /.cache/gorilla', shell=True, capture_output=True, text=True)
|
34 |
log(result.stdout)
|
35 |
|
36 |
-
#model_id : str = "gorilla-llm/gorilla-openfunctions-v1"
|
37 |
-
model_id : str = "M-FAC/bert-tiny-finetuned-mrpc"
|
38 |
log('AutoTokenizer.from_pretrained ...')
|
39 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
40 |
log('AutoModelForCausalLM.from_pretrained ...')
|
|
|
33 |
result = subprocess.run('pwd && ls -lH && find /.cache/huggingface/hub && find /.cache/gorilla', shell=True, capture_output=True, text=True)
|
34 |
log(result.stdout)
|
35 |
|
|
|
|
|
36 |
log('AutoTokenizer.from_pretrained ...')
|
37 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
38 |
log('AutoModelForCausalLM.from_pretrained ...')
|