WallyLovesCats commited on
Commit
d6f1ffa
·
verified ·
1 Parent(s): bd069b7

Add files using upload-large-folder tool

Browse files
Files changed (2) hide show
  1. .gitattributes +0 -0
  2. run_nvila.py +2 -2
.gitattributes CHANGED
The diff for this file is too large to render. See raw diff
 
run_nvila.py CHANGED
@@ -9,9 +9,9 @@ quantization_config = BitsAndBytesConfig(
9
  bnb_4bit_use_double_quant=True,
10
  )
11
 
12
- tokenizer = AutoTokenizer.from_pretrained("../NVILA-Lite-8B/llm")
13
  model = AutoModelForCausalLM.from_pretrained(
14
- "../NVILA-Lite-8B/llm",
15
  torch_dtype=torch.bfloat16,
16
  device_map="auto",
17
  quantization_config=quantization_config
 
9
  bnb_4bit_use_double_quant=True,
10
  )
11
 
12
+ tokenizer = AutoTokenizer.from_pretrained("WallyLovesCats/NVILA-Lite-8B/llm")
13
  model = AutoModelForCausalLM.from_pretrained(
14
+ "WallyLovesCats/NVILA-Lite-8B/llm",
15
  torch_dtype=torch.bfloat16,
16
  device_map="auto",
17
  quantization_config=quantization_config