aizanlabs commited on
Commit
64f4a33
·
verified ·
1 Parent(s): 2d1ac8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -71,8 +71,8 @@ class DocumentRetrievalAndGeneration:
71
  return generate_text
72
  def initialize_llm2(self,model_id):
73
 
74
- tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-1b3", local_files_only=True)
75
- model = BloomForCausalLM.from_pretrained("bigscience/bloom-1b3", local_files_only=True)
76
  result_length = 200
77
  inputs = tokenizer(prompt, return_tensors="pt")
78
  # return generate_text
 
71
  return generate_text
72
  def initialize_llm2(self,model_id):
73
 
74
+ tokenizer = BloomTokenizerFast.from_pretrained("bigscience/bloom-1b3")#, local_files_only=True)
75
+ model = BloomForCausalLM.from_pretrained("bigscience/bloom-1b3")#, local_files_only=True)
76
  result_length = 200
77
  inputs = tokenizer(prompt, return_tensors="pt")
78
  # return generate_text