Commit 856bd31
minymal llama backbone export (pytorch#8453)
Summary:
Export llama 1b from minymal (for testing purpose fixed shape of 1*len*2048) is used. This is build on based on the llava example
Reviewed By: asleep, mcr229
Differential Revision: D695729541 parent 64b0fa7 commit 856bd31
1 file changed
+4
-1
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
47 | 47 | | |
48 | 48 | | |
49 | 49 | | |
50 | | - | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
51 | 54 | | |
52 | 55 | | |
53 | 56 | | |
| |||
0 commit comments