ffreemt commited on
Commit
d798dc0
·
1 Parent(s): 2321903
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -44,7 +44,9 @@ else:
44
  try:
45
  # model = AutoModel.from_pretrained(model_name, trust_remote_code=True).float()
46
  model = AutoModelForCausalLM.from_pretrained(
47
- model_name, trust_remote_code=True
 
 
48
  ).float()
49
  except Exception as exc:
50
  logger.error(exc)
 
44
  try:
45
  # model = AutoModel.from_pretrained(model_name, trust_remote_code=True).float()
46
  model = AutoModelForCausalLM.from_pretrained(
47
+ # model_name, trust_remote_code=True
48
+ "model",
49
+ trust_remote_code=True,
50
  ).float()
51
  except Exception as exc:
52
  logger.error(exc)