Update app.py

This commit is contained in:
Baize 2023-04-02 02:57:24 +00:00 committed by huggingface-web
parent 8232000a59
commit b2cae66391
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,