Hugging Face's logo

Spaces:
Akjava
/
MiniCPM-2B-CPU
Running

App Files Community
Akjava commited on
Commit
f4ffb0a
·
1 Parent(s): fd91f3c

update

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -71,7 +71,7 @@ iface = gr.Interface(
71
  print("Initialized")
72
 
73
  model = AutoModelForCausalLM.from_pretrained(
74
- model_id, token=huggingface_token ,torch_dtype=dtype,device_map=device
75
  )
76
 
77
  def generate_text(messages):
 
71
  print("Initialized")
72
 
73
  model = AutoModelForCausalLM.from_pretrained(
74
+ model_id, token=huggingface_token ,torch_dtype=dtype,device_map=device,trust_remote_code=True
75
  )
76
 
77
  def generate_text(messages):