Commit ·
3c0b28f
1
Parent(s): 16a6214
Update README.md
Browse files
README.md
CHANGED
|
@@ -35,17 +35,12 @@ class ISPInConfig(PretrainedConfig):
|
|
| 35 |
Python code for ISPIn.py
|
| 36 |
```python
|
| 37 |
import gc
|
| 38 |
-
|
| 39 |
import numpy as np
|
| 40 |
-
|
| 41 |
import torch
|
| 42 |
from torch import nn
|
| 43 |
-
|
| 44 |
from tqdm.notebook import tqdm
|
| 45 |
from tqdm.autonotebook import trange
|
| 46 |
-
|
| 47 |
from transformers import PreTrainedModel, AutoTokenizer, AutoModel
|
| 48 |
-
|
| 49 |
from .ISPInConfig import ISPInConfig
|
| 50 |
|
| 51 |
class ISPIn(PreTrainedModel):
|
|
|
|
| 35 |
Python code for ISPIn.py
|
| 36 |
```python
|
| 37 |
import gc
|
|
|
|
| 38 |
import numpy as np
|
|
|
|
| 39 |
import torch
|
| 40 |
from torch import nn
|
|
|
|
| 41 |
from tqdm.notebook import tqdm
|
| 42 |
from tqdm.autonotebook import trange
|
|
|
|
| 43 |
from transformers import PreTrainedModel, AutoTokenizer, AutoModel
|
|
|
|
| 44 |
from .ISPInConfig import ISPInConfig
|
| 45 |
|
| 46 |
class ISPIn(PreTrainedModel):
|