Update README.md
Browse files
README.md
CHANGED
|
@@ -26,7 +26,7 @@ The models in the LH-Tech AI Pin Series are very small models that were trained
|
|
| 26 |
| Pin-15M | 14.84M | 1500 | 2.358367 | Low | It is hard to see your plans when you want to enjoy the day. |
|
| 27 |
| Pin-20M | 21.03M | 1500 | 2.217588 | Medium | Yes, sunny days are wonderful! I love hearing about the sunshine and the sun's shining on. |
|
| 28 |
| Pin-25M | 26.76M | 1500 | 2.139837 | Medium | Sunny days make everything look brighter, especially with a nice friend who cares. |
|
| 29 |
-
| Pin-Ultra-25M | 26.76M | 8000 | .
|
| 30 |
|
| 31 |
\* All models were prompted with `What is the weather like today?`.
|
| 32 |
|
|
@@ -51,7 +51,7 @@ You can find the full training code for the Pin Model Series in this repo.<br>
|
|
| 51 |
Have fun :D
|
| 52 |
|
| 53 |
### Training details
|
| 54 |
-
We trained all these models in ~30 minutes on
|
| 55 |
So you are able to easily recreate all of the Pin model without having to launch a 8xH100 cluster 😂
|
| 56 |
|
| 57 |
## How to use the model
|
|
@@ -60,7 +60,7 @@ You can easily use the favorite model of the Pin series like this:
|
|
| 60 |
1. Download `use.py` from this repo.
|
| 61 |
2. Adjust the subfolder argument here:
|
| 62 |
```python
|
| 63 |
-
answer = run_pin_inference(user_query, model_id="LH-Tech-AI/Pin", subfolder="Pin-25M") # use your favorite model here, e.g. "Pin-25M" or "Pin-15M"...
|
| 64 |
```
|
| 65 |
3. Adjust the input prompt here:
|
| 66 |
```python
|
|
@@ -73,5 +73,5 @@ You can easily use the favorite model of the Pin series like this:
|
|
| 73 |
...to:
|
| 74 |
|
| 75 |
- HF Transformers
|
| 76 |
-
- Kaggle for the
|
| 77 |
- starhopp3r for his TinyChat dataset
|
|
|
|
| 26 |
| Pin-15M | 14.84M | 1500 | 2.358367 | Low | It is hard to see your plans when you want to enjoy the day. |
|
| 27 |
| Pin-20M | 21.03M | 1500 | 2.217588 | Medium | Yes, sunny days are wonderful! I love hearing about the sunshine and the sun's shining on. |
|
| 28 |
| Pin-25M | 26.76M | 1500 | 2.139837 | Medium | Sunny days make everything look brighter, especially with a nice friend who cares. |
|
| 29 |
+
| Pin-Ultra-25M | 26.76M | 8000 | 1.588281 | Hight | Yes, I spent some time at the park and was looking for a new collar. |
|
| 30 |
|
| 31 |
\* All models were prompted with `What is the weather like today?`.
|
| 32 |
|
|
|
|
| 51 |
Have fun :D
|
| 52 |
|
| 53 |
### Training details
|
| 54 |
+
We trained all these models in ~30 minutes on Kaggle 2xT4 GPU in a Kaggle Session (except for Pin-Ultra-25 - it trained for ~4 hours on Kaggle 2xT4).<br>
|
| 55 |
So you are able to easily recreate all of the Pin model without having to launch a 8xH100 cluster 😂
|
| 56 |
|
| 57 |
## How to use the model
|
|
|
|
| 60 |
1. Download `use.py` from this repo.
|
| 61 |
2. Adjust the subfolder argument here:
|
| 62 |
```python
|
| 63 |
+
answer = run_pin_inference(user_query, model_id="LH-Tech-AI/Pin", subfolder="Pin-Ultra-25M") # use your favorite model here, e.g. "Pin-25M" or "Pin-15M"...
|
| 64 |
```
|
| 65 |
3. Adjust the input prompt here:
|
| 66 |
```python
|
|
|
|
| 73 |
...to:
|
| 74 |
|
| 75 |
- HF Transformers
|
| 76 |
+
- Kaggle for the 2xT4 GPU
|
| 77 |
- starhopp3r for his TinyChat dataset
|