SoumyaJ commited on
Commit
ef41f4d
1 Parent(s): 90f39f3

Update src/TinyLLama/text_generation.py

Browse files
Files changed (1) hide show
  1. src/TinyLLama/text_generation.py +1 -1
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