Deepseek Math 7b Base

🚀 DeepSeekMath

DeepSeekMath是一款强大的数学相关模型，可用于文本补全，为数学问题的解决提供支持，助力数学研究与应用。

🚀 快速开始

若想了解更多关于DeepSeekMath的详细信息，请查看介绍。

💻 使用示例

基础用法

import torch
from transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfig

model_name = "deepseek-ai/deepseek-math-7b-base"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
model.generation_config = GenerationConfig.from_pretrained(model_name)
model.generation_config.pad_token_id = model.generation_config.eos_token_id

text = "The integral of x^2 from 0 to 2 is"
inputs = tokenizer(text, return_tensors="pt")
outputs = model.generate(**inputs.to(model.device), max_new_tokens=100)

result = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(result)

📄 许可证

此代码仓库遵循MIT许可证。DeepSeekMath模型的使用需遵循模型许可证，且DeepSeekMath支持商业使用。

更多详细信息请查看LICENSE-MODEL。

📞 联系我们

如果您有任何问题，请提出issue或通过service@deepseek.com联系我们。

项目相关链接

[🏠主页] | [🤖 与DeepSeek LLM聊天] | [Discord] | [微信(Wechat)]

论文链接👁️