jiachengcui888 commited on
Commit
717ec31
·
verified ·
1 Parent(s): e6fb316

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -18
README.md CHANGED
@@ -52,24 +52,7 @@ PIXAR-7B achieves **2.6× IoU improvement** over prior state of the art on the P
52
 
53
  ## How to Get Started
54
 
55
- ```python
56
- from transformers import AutoTokenizer
57
- from model.PIXAR import PIXARForCausalLM
58
-
59
- model = PIXARForCausalLM.from_pretrained(
60
- "jiachengcui888/PIXAR-7B",
61
- low_cpu_mem_usage=True,
62
- vision_tower="openai/clip-vit-large-patch14",
63
- seg_token_idx=seg_token_idx,
64
- cls_token_idx=cls_token_idx,
65
- obj_token_idx=obj_token_idx,
66
- num_obj_classes=81,
67
- seg_prompt_mode="seg_only",
68
- torch_dtype=torch.bfloat16,
69
- ).cuda()
70
- ```
71
-
72
- For interactive inference, see the [project repository](https://github.com/Jiacheng8/PIXAR):
73
 
74
  ```bash
75
  python chat.py --version jiachengcui888/PIXAR-7B --precision bf16 --seg_prompt_mode seg_only
 
52
 
53
  ## How to Get Started
54
 
55
+ For interactive inference, see the [project repository](https://github.com/VILA-Lab/PIXAR):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
56
 
57
  ```bash
58
  python chat.py --version jiachengcui888/PIXAR-7B --precision bf16 --seg_prompt_mode seg_only