jiajunlong
/

TinyLLaVA-OpenELM-450M-SigLIP-0.89B

Image-Text-to-Text

text-generation

Model card Files Files and versions Community

jiajunlong commited on Apr 29

Commit

f351ec3

•

1 Parent(s): a48fa5a

Update generate_model.py

Files changed (1) hide show

generate_model.py +0 -1

generate_model.py CHANGED Viewed

@@ -589,7 +589,6 @@ def generate(
     # print(f'loading model from {checkpoint_path}...')
         model = AutoModelForCausalLM.from_pretrained(
             checkpoint_path,
-            torch_dtype=torch.float16,
             trust_remote_code=True
         )
     # print('model load over')

     # print(f'loading model from {checkpoint_path}...')
         model = AutoModelForCausalLM.from_pretrained(
             checkpoint_path,
             trust_remote_code=True
         )
     # print('model load over')