Spaces:

Akjava
/

MiniCPM-2B-CPU

Runtime error

Akjava commited on Aug 5, 2024

Commit

fd91f3c

1 Parent(s): 04e5516

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model_id = "openbmb/MiniCPM-2B-dpo-bf16"
 device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dtype = torch.bfloat16
-tokenizer = AutoTokenizer.from_pretrained(model_id, token=huggingface_token)
 print(model_id,device,dtype)
 histories = []

 device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
 dtype = torch.bfloat16
+tokenizer = AutoTokenizer.from_pretrained(model_id,trust_remote_code=True)
 print(model_id,device,dtype)
 histories = []