Hugging Face's logo

Spaces:
ActiveYixiao
/
automatic_coding
Sleeping

App Files Community
Yixiao Wang (Computer Science) commited on
Commit
82800a1
·
1 Parent(s): cd15e92

change to longformer

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from transformers import AutoModelForCausalLM, AutoModelForSequenceClassificatio
14
  logging.basicConfig(level=logging.INFO)
15
  logger = logging.getLogger(__name__)
16
 
17
- MODEL_ID = "rshwndsz/ft-hermes-3-llama-3.2-3b"
18
  DEVICE_MAP = "auto"
19
  QUANTIZATION_BITS = None
20
  TEMPERATURE = 0.0
 
14
  logging.basicConfig(level=logging.INFO)
15
  logger = logging.getLogger(__name__)
16
 
17
+ MODEL_ID = "rshwndsz/ft-longformer-base-4096"
18
  DEVICE_MAP = "auto"
19
  QUANTIZATION_BITS = None
20
  TEMPERATURE = 0.0