zxx_test
/
app_demo_test_1
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Update app.py

This commit is contained in:
Baize 2023-04-02 03:48:29 +00:00 committed by huggingface-web
parent ca62df39ae
commit 1a2243e5f5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@ -15,7 +15,7 @@ logging.basicConfig(
base_model = "decapoda-research/llama-7b-hf" base_model = "decapoda-research/llama-7b-hf"
adapter_model = "project-baize/baize-lora-7B" adapter_model = "project-baize/baize-lora-7B"
tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model) #tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
def predict(text, def predict(text,