Krutrim-2-instruct

Running on Zero

eswardivi commited on 17 days ago

Commit

90a750e

verified ·

1 Parent(s): a6a230c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True,
     torch_dtype=torch.bfloat16
 )
-tok = AutoTokenizer.from_pretrained("krutrim-ai-labs/Krutrim-2-instruct", token=token)
 terminators = [
     tok.eos_token_id,
 ]

     trust_remote_code=True,
     torch_dtype=torch.bfloat16
 )
+tok = AutoTokenizer.from_pretrained("krutrim-ai-labs/Krutrim-2-instruct")
 terminators = [
     tok.eos_token_id,
 ]