Update app.py
This commit is contained in:
parent
a56a74d5e6
commit
fdf1396367
2
app.py
2
app.py
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue