Spaces:
Runtime error
Runtime error
Commit
·
7ffb3a3
1
Parent(s):
ca799f1
Update
Browse files- run_llm.py +2 -0
run_llm.py
CHANGED
|
@@ -9,6 +9,8 @@ import requests
|
|
| 9 |
from tqdm import tqdm
|
| 10 |
import torch
|
| 11 |
from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaForCausalLM, LlamaTokenizer
|
|
|
|
|
|
|
| 12 |
|
| 13 |
from fastchat.model import load_model, get_conversation_template, add_model_args
|
| 14 |
|
|
|
|
| 9 |
from tqdm import tqdm
|
| 10 |
import torch
|
| 11 |
from transformers import AutoTokenizer, AutoModelForCausalLM, LlamaForCausalLM, LlamaTokenizer
|
| 12 |
+
from huggingface_hub import HfApi
|
| 13 |
+
api = HfApi()
|
| 14 |
|
| 15 |
from fastchat.model import load_model, get_conversation_template, add_model_args
|
| 16 |
|