updata inference

72fa11c7 · shihm · 61f51a8e · 72fa11c7
Commit 72fa11c7 authored Mar 13, 2026 by shihm
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

inference.py inference.py +1 -1

No files found.
--- a/inference.py
+++ b/inference.py
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import os
 import torch
-model_path = "/baichuan-inc/Baichuan-M3-235B"
+model_path = "baichuan-inc/Baichuan-M3-235B"
 os.environ['TRANSFORMERS_OFFLINE'] = '1'
 os.environ['MODELSCOPE_OFFLINE'] = '1'
 model = AutoModelForCausalLM.from_pretrained(