nikravan commited on
Commit
62ebda0
·
verified ·
1 Parent(s): aa3fb34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -111,7 +111,7 @@ quantization_config = BitsAndBytesConfig(
111
  load_in_4bit=True,
112
  bnb_4bit_compute_dtype=torch.bfloat16
113
  )
114
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
115
  model = AutoModelForCausalLM.from_pretrained(
116
  MODEL_ID,
117
  device_map="auto",
 
111
  load_in_4bit=True,
112
  bnb_4bit_compute_dtype=torch.bfloat16
113
  )
114
+ tokenizer = AutoTokenizer.from_pretrained('AIDC-AI/Marco-o1')
115
  model = AutoModelForCausalLM.from_pretrained(
116
  MODEL_ID,
117
  device_map="auto",