Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|