Singularity666 commited on
Commit
afa2fb8
·
verified ·
1 Parent(s): 17a5ce9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -330,4 +330,4 @@ class LlavaLlamaForCausalLM(LlamaForCausalLM):
330
  vision_config.im_patch_token = tokenizer.convert_tokens_to_ids([DEFAULT_IMAGE_PATCH_TOKEN])[0]
331
 
332
  AutoConfig.register("llava", LlavaConfig)
333
- AutoModelForCausalLM.register(LlavaConfig, LlavaLlamaForCausalLM)
 
330
  vision_config.im_patch_token = tokenizer.convert_tokens_to_ids([DEFAULT_IMAGE_PATCH_TOKEN])[0]
331
 
332
  AutoConfig.register("llava", LlavaConfig)
333
+ AutoModelForCausalLM.register(LlavaConfig, LlavaLlamaForCausalLM)