Update app.py

This commit is contained in:
Baize 2023-04-02 03:38:43 +00:00 committed by huggingface-web
parent 9a6c79301d
commit ca62df39ae
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" 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,