Hjgugugjhuhjggg commited on
Commit
1692af9
·
verified ·
1 Parent(s): 37f7b00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ async def predict(request: DownloadModelRequest):
110
  with open(model_path, 'wb') as f:
111
  f.write(model_stream.read())
112
 
113
- model = AutoModelForCausalLM.from_pretrained(tmp_dir, from_tf=False)
114
  tokenizer = AutoTokenizer.from_pretrained(tmp_dir)
115
 
116
  pipeline_ = pipeline(request.pipeline_task, model=model, tokenizer=tokenizer)
 
110
  with open(model_path, 'wb') as f:
111
  f.write(model_stream.read())
112
 
113
+ model = AutoModelForCausalLM.from_pretrained(tmp_dir, from_tf=True)
114
  tokenizer = AutoTokenizer.from_pretrained(tmp_dir)
115
 
116
  pipeline_ = pipeline(request.pipeline_task, model=model, tokenizer=tokenizer)