ysharma HF staff commited on
Commit
6a5acea
·
verified ·
1 Parent(s): 42493a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -48,7 +48,10 @@ model_id = "tiiuae/falcon-11B-vlm"
48
  #processor = AutoProcessor.from_pretrained(model_id)
49
  processor = LlavaNextProcessor.from_pretrained("tiiuae/falcon-11B-vlm", tokenizer_class='PreTrainedTokenizerFast')
50
 
51
- model = LlavaNextForConditionalGeneration.from_pretrained("tiiuae/falcon-11B-vlm", torch_dtype=torch.bfloat16)
 
 
 
52
 
53
  #model = LlavaForConditionalGeneration.from_pretrained(
54
  # model_id,
 
48
  #processor = AutoProcessor.from_pretrained(model_id)
49
  processor = LlavaNextProcessor.from_pretrained("tiiuae/falcon-11B-vlm", tokenizer_class='PreTrainedTokenizerFast')
50
 
51
+ model = LlavaNextForConditionalGeneration.from_pretrained("tiiuae/falcon-11B-vlm",
52
+ #torch_dtype=torch.bfloat16,
53
+ torch_dtype=torch.float16,
54
+ low_cpu_mem_usage=True,)
55
 
56
  #model = LlavaForConditionalGeneration.from_pretrained(
57
  # model_id,