ciyidogan commited on
Commit
03f9f6e
·
verified ·
1 Parent(s): 8f06ef3

Update llm_model.py

Browse files
Files changed (1) hide show
  1. llm_model.py +1 -1
llm_model.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch, traceback
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
3
- import log
4
  from pydantic import BaseModel
5
 
6
  global model, tokenizer, eos_token_id
 
1
  import torch, traceback
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
3
+ from log import log
4
  from pydantic import BaseModel
5
 
6
  global model, tokenizer, eos_token_id