KangarooGroup
/

kangaroo

Video-Text-to-Text

text-generation

text-generation-inference

Model card Files Files and versions

WEBing commited on Jul 22, 2024

Commit

3da7b6a

·

1 Parent(s): 9eeb018

update readme

Files changed (1) hide show

README.md +2 -3

README.md CHANGED Viewed

@@ -30,10 +30,9 @@ See our [github page](https://github.com/KangarooGroup/Kangaroo)
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
-model_path = "/path/to/kangaroo"
-tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(
-    model_path,
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
 )

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+tokenizer = AutoTokenizer.from_pretrained("KangarooGroup/kangaroo")
 model = AutoModelForCausalLM.from_pretrained(
+    "KangarooGroup/kangaroo",
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
 )