Commit ·
935f2f3
1
Parent(s): c860158
fix readme instruction
Browse files
README.md
CHANGED
|
@@ -24,7 +24,10 @@ You can try out the MetaLATTE model directly in your browser:
|
|
| 24 |
## Usage
|
| 25 |
|
| 26 |
```python
|
|
|
|
| 27 |
from transformers import AutoTokenizer, AutoModel, AutoConfig
|
|
|
|
|
|
|
| 28 |
from metalatte import MetaLATTEConfig, MetaLATTEModel
|
| 29 |
AutoConfig.register("metalatte", MetaLATTEConfig)
|
| 30 |
AutoModel.register(MetaLATTEConfig, MetaLATTEModel)
|
|
|
|
| 24 |
## Usage
|
| 25 |
|
| 26 |
```python
|
| 27 |
+
import sys
|
| 28 |
from transformers import AutoTokenizer, AutoModel, AutoConfig
|
| 29 |
+
metalatte_path = './Chatterjeelab/MetaLATTE'
|
| 30 |
+
sys.path.insert(0, metalatte_path)
|
| 31 |
from metalatte import MetaLATTEConfig, MetaLATTEModel
|
| 32 |
AutoConfig.register("metalatte", MetaLATTEConfig)
|
| 33 |
AutoModel.register(MetaLATTEConfig, MetaLATTEModel)
|