Spaces:

dennyaw
/

polylm-1.7b

Sleeping

dennyaw commited on Oct 18, 2023

Commit

830c2e2

•

1 Parent(s): 02c6b08

update import peft

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 model = AutoModelForCausalLM.from_pretrained("DAMO-NLP-MT/polylm-1.7b")
 model = PeftModel.from_pretrained(model, "fadliaulawi/polylm-1.7b-finetuned")

 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+from peft import PeftModel
 model = AutoModelForCausalLM.from_pretrained("DAMO-NLP-MT/polylm-1.7b")
 model = PeftModel.from_pretrained(model, "fadliaulawi/polylm-1.7b-finetuned")