lxz8798 commited on
Commit
5b71e1d
·
verified ·
1 Parent(s): cfae321

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +16 -1
handler.py CHANGED
@@ -1,9 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import torch
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
 
4
  class EndpointHandler:
5
  def __init__(self, path=""):
6
- # 1. 强制使用 trust_remote_code=True 加载 Tokenizer 和 Model
7
  self.tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True)
8
  self.model = AutoModelForCausalLM.from_pretrained(
9
  path,
 
1
+ import subprocess
2
+ import sys
3
+
4
+ # 🚀 强制在代码运行最开始,从 PyPI 热升级 transformers 和 accelerate 到最新版本
5
+ try:
6
+ print("Starting hot-upgrade of transformers and accelerate...")
7
+ subprocess.run(
8
+ [sys.executable, "-m", "pip", "install", "--upgrade", "transformers", "accelerate", "--user"],
9
+ check=True
10
+ )
11
+ print("Hot-upgrade completed successfully!")
12
+ except Exception as e:
13
+ print(f"Warning during hot-upgrade: {e}")
14
+
15
+ # ----------------- 升级完成后再导入库 -----------------
16
  import torch
17
  from transformers import AutoModelForCausalLM, AutoTokenizer
18
 
19
  class EndpointHandler:
20
  def __init__(self, path=""):
21
+ # 1. 使用 trust_remote_code=True 加载 Tokenizer 和 Model
22
  self.tokenizer = AutoTokenizer.from_pretrained(path, trust_remote_code=True)
23
  self.model = AutoModelForCausalLM.from_pretrained(
24
  path,