Spaces:
Sleeping
Sleeping
Update src/TinyLLama/text_generation.py
Browse files
src/TinyLLama/text_generation.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
2 |
from src.classmodels.inputforgeneration import InputForGeneration
|
3 |
-
from errorlog.errorlog import log_error
|
4 |
from pathlib import Path
|
5 |
|
6 |
#MODEL NAME AS PER IN HUGGING FACE :- TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
|
|
|
1 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
2 |
from src.classmodels.inputforgeneration import InputForGeneration
|
3 |
+
from src.errorlog.errorlog import log_error
|
4 |
from pathlib import Path
|
5 |
|
6 |
#MODEL NAME AS PER IN HUGGING FACE :- TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T
|