Fix

Add trust_remote_code

README.md

@@ -42,7 +42,7 @@ Note that no additional fine-tuning has been performed on this model. As such, i
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained("OxxoCodes/jamba-small-v2", torch_dtype=torch.bfloat16)
+model = AutoModelForCausalLM.from_pretrained("OxxoCodes/jamba-small-v2", torch_dtype=torch.bfloat16, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained("ai21labs/Jamba-v0.1")
 
 with torch.no_grad():