Spaces:

Akjava
/

MiniCPM-2B-CPU

Sleeping

Akjava commited on Aug 5

Commit

f4ffb0a

•

1 Parent(s): fd91f3c

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -71,7 +71,7 @@ iface = gr.Interface(
 print("Initialized")
 model = AutoModelForCausalLM.from_pretrained(
-        model_id, token=huggingface_token ,torch_dtype=dtype,device_map=device
     )
 def generate_text(messages):

 print("Initialized")
 model = AutoModelForCausalLM.from_pretrained(
+        model_id, token=huggingface_token ,torch_dtype=dtype,device_map=device,trust_remote_code=True
     )
 def generate_text(messages):