Ramikan-BR
commited on
Commit
•
3c11e04
1
Parent(s):
e77bc45
Create generate_config.py
Browse filesEsse código cria o config.json da minha IA treinada.
- generate_config.py +7 -0
generate_config.py
ADDED
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import AutoConfig, LlamaForCausalLM
|
2 |
+
|
3 |
+
model = LlamaForCausalLM.from_pretrained("Ramikan-BR/tinyllama_PY-CODER-bnb-4bit-lora_model-4k")
|
4 |
+
config = model.config
|
5 |
+
|
6 |
+
config_dict = config.to_dict()
|
7 |
+
print(config_dict)
|