liltom-eth commited on
Commit
d2cca8d
·
1 Parent(s): 4958310

update inference

Browse files
Files changed (2) hide show
  1. code/inference.py +1 -1
  2. code/requirements.txt +0 -1
code/inference.py CHANGED
@@ -2,7 +2,7 @@ import requests
2
  from PIL import Image
3
  from io import BytesIO
4
  import torch
5
- from transformers import AutoTokenizer, BitsAndBytesConfig
6
 
7
  from llava.model import LlavaLlamaForCausalLM
8
  from llava.utils import disable_torch_init
 
2
  from PIL import Image
3
  from io import BytesIO
4
  import torch
5
+ from transformers import AutoTokenizer
6
 
7
  from llava.model import LlavaLlamaForCausalLM
8
  from llava.utils import disable_torch_init
code/requirements.txt CHANGED
@@ -1,2 +1 @@
1
- # llava @ git+https://github.com/haotian-liu/LLaVA@main
2
  llava @ git+https://github.com/haotian-liu/[email protected]
 
 
1
  llava @ git+https://github.com/haotian-liu/[email protected]