Update app.py

This commit is contained in:
Baize 2023-04-02 05:13:35 +00:00 committed by huggingface-web
parent a56a74d5e6
commit fdf1396367
1 changed files with 1 additions and 1 deletions

2
app.py
View File

@ -15,7 +15,7 @@ logging.basicConfig(
base_model = "decapoda-research/llama-7b-hf"
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,