EPlus-LLM commited on
Commit
786baea
·
verified ·
1 Parent(s): e833828

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
README.md CHANGED
@@ -46,10 +46,6 @@ from transformers import (
46
  AutoTokenizer,
47
  )
48
 
49
- # Load the rest port of IDF file.
50
- file_path = "v1_nextpart.idf" # File is in the repo, please download.
51
- output_path = "v1_final.idf"
52
-
53
  # Load the EPlus-LLM model
54
  tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
55
  model = AutoModelForSeq2SeqLM.from_pretrained("EPlus-LLM/EPlus-LLMv1"
@@ -76,6 +72,9 @@ generated_output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
76
  generated_output = generated_output.replace("_", " ")
77
  generated_output = generated_output.replace("|", "\n")
78
 
 
 
 
79
  with open(file_path, 'r', encoding='utf-8') as file:
80
  nextpart = file.read()
81
  final_text = nextpart + "\n\n" + generated_output
 
46
  AutoTokenizer,
47
  )
48
 
 
 
 
 
49
  # Load the EPlus-LLM model
50
  tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
51
  model = AutoModelForSeq2SeqLM.from_pretrained("EPlus-LLM/EPlus-LLMv1"
 
72
  generated_output = generated_output.replace("_", " ")
73
  generated_output = generated_output.replace("|", "\n")
74
 
75
+ # Load the rest port of IDF file.
76
+ file_path = "v1_nextpart.idf" # File is in the repo, please download.
77
+ output_path = "v1_final.idf"
78
  with open(file_path, 'r', encoding='utf-8') as file:
79
  nextpart = file.read()
80
  final_text = nextpart + "\n\n" + generated_output