LH-Tech-AI commited on
Commit
5c0f838
·
verified ·
1 Parent(s): a2de7cb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
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 | ... | ... | Coming soon... |
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 a single T4 GPU in a Kaggle Session.<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,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 T4 GPU
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