LiquidAI
/

LFM2-700M

Text Generation

Model card Files Files and versions

mlabonne commited on 5 days ago

Commit

c0e030b

·

verified ·

1 Parent(s): 85efa52

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -155,7 +155,6 @@ model = AutoModelForCausalLM.from_pretrained(
     model_id,
     device_map="auto",
     torch_dtype="bfloat16",
-    trust_remote_code=True,
 #    attn_implementation="flash_attention_2" <- uncomment on compatible GPU
 )
 tokenizer = AutoTokenizer.from_pretrained(model_id)

     model_id,
     device_map="auto",
     torch_dtype="bfloat16",
 #    attn_implementation="flash_attention_2" <- uncomment on compatible GPU
 )
 tokenizer = AutoTokenizer.from_pretrained(model_id)