Skip to content

Commit

Permalink
change model
Browse files Browse the repository at this point in the history
  • Loading branch information
garylvov committed Oct 11, 2024
1 parent df2b1ff commit a490961
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -262,7 +262,7 @@ def __init__(
"TheiaTiny": {
"model": (
lambda: AutoModel.from_pretrained(
"theaiinstitute/theia-tiny-patch16-224-cdiv", trust_remote_code=True
"theaiinstitute/theia-tiny-patch16-224-cddsv", trust_remote_code=True
)
.eval()
.to(device)
Expand Down

0 comments on commit a490961

Please sign in to comment.