Skip to content

Commit e2b1358

Browse files
Add back trocr removed in last commit
1 parent 3b2bcd6 commit e2b1358

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

src/Models/Auto/AutoModelForCausalLM.php

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ class AutoModelForCausalLM extends PretrainedMixin
1212
'gptj' => \Codewithkyrian\Transformers\Models\Pretrained\GPTJForCausalLM::class,
1313
'gpt_bigcode' => \Codewithkyrian\Transformers\Models\Pretrained\GPTBigCodeForCausalLM::class,
1414
'codegen' => \Codewithkyrian\Transformers\Models\Pretrained\CodeGenForCausalLM::class,
15+
'trocr' => \Codewithkyrian\Transformers\Models\Pretrained\TrOCRForCausalLM::class,
1516
'qwen2' => \Codewithkyrian\Transformers\Models\Pretrained\Qwen2ForCausalLM::class
1617
];
1718

0 commit comments

Comments
 (0)