Spaces:
Runtime error
Runtime error
gmerrill
commited on
Commit
·
efec77b
1
Parent(s):
085d5a7
update
Browse files
main.py
CHANGED
@@ -33,6 +33,7 @@ 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 |
log('AutoTokenizer.from_pretrained ...')
|
37 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
38 |
log('AutoModelForCausalLM.from_pretrained ...')
|
@@ -82,4 +83,3 @@ def index() -> FileResponse:
|
|
82 |
|
83 |
|
84 |
log('Initialization done.')
|
85 |
-
|
|
|
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 |
log('AutoTokenizer.from_pretrained ...')
|
38 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
39 |
log('AutoModelForCausalLM.from_pretrained ...')
|
|
|
83 |
|
84 |
|
85 |
log('Initialization done.')
|
|