Skip to content

Commit

Permalink
Add LoRA option for VisionTransformer (#3729)
Browse files Browse the repository at this point in the history 
* replacement with timm vit

* update vit with timm

* fix explain for vit

* add unit test

* enable more models

* add v1 checkponit loader

* restore pretrained weight

* replace with suffix

* tmp

* tmp2

* refactor vit

* lora arg

* restore
  • Loading branch information
@wonjuleee
wonjuleee authored Jul 16, 2024
1 parent 2b00009 commit 30bcc40
Show file tree
Hide file tree
Showing 3 changed files with 262 additions and 128 deletions.
Loading

0 comments on commit 30bcc40

Please sign in to comment.