Upload pytorch_model.bin

# test_load_pytorch_model.py
# ✅ 测试 Hugging Face 标准格式 pytorch_model.bin 是否能被正确加载

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
import os

# ---------- 模型目录 ----------
# 假设当前目录包含 config.json 和 pytorch_model.bin
model_dir = "./"

# ---------- 加载模型 ----------
print("🚀 正在加载模型...")
model = AutoModelForCausalLM.from_pretrained(
model_dir,
torch_dtype="auto"
)
print("✅ 模型加载成功")

# ---------- 加载分词器 ----------
tokenizer = AutoTokenizer.from_pretrained(model_dir)

# ---------- 推理测试 ----------
prompt = "你好，今天星期几？"
inputs = tokenizer(prompt, return_tensors="pt")
with torch.no_grad():
outputs = model.generate(
**inputs,
max_new_tokens=50,
do_sample=True,
top_p=0.9,
temperature=0.7
)

response = tokenizer.decode(outputs[0], skip_special_tokens=True)
print("🤖 模型回复：")
print(response)

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e4360068a39d5ec970c37925a20e39368c4d2624861ce6f873b9397d813edd25
+size 1043