credent007 commited on
Commit
47de079
·
verified ·
1 Parent(s): 9d0c20f

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +4 -3
llm.py CHANGED
@@ -1,12 +1,13 @@
1
- import torch
 
2
  import asyncio
3
  from functools import partial
4
  import time
5
  from transformers import AutoProcessor, AutoModelForImageTextToText #, BitsAndBytesConfig
6
- import os
7
  # Quantization config
8
  # quant_config = BitsAndBytesConfig(load_in_8bit=True)
9
- hf_token=os.getenv("HUGGINGFACE_HUB_TOKEN")
10
  # Load processor
11
  processor = AutoProcessor.from_pretrained("datalab-to/chandra-ocr-2")
12
 
 
1
+ import os
2
+ hf_token=os.getenv("HUGGINGFACE_HUB_TOKEN")import torch
3
  import asyncio
4
  from functools import partial
5
  import time
6
  from transformers import AutoProcessor, AutoModelForImageTextToText #, BitsAndBytesConfig
7
+
8
  # Quantization config
9
  # quant_config = BitsAndBytesConfig(load_in_8bit=True)
10
+
11
  # Load processor
12
  processor = AutoProcessor.from_pretrained("datalab-to/chandra-ocr-2")
13