Spaces:
Runtime error
Runtime error
Singularity666
commited on
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)
|