原標題：清華系「自然語言編程神器」上新！支持100+種編程語言，效率upup

克雷西 發自 凹非寺

量子位 | 公眾號 QbitAI

GitHub Copilot的模型也不是對手

快速體驗

transformers importAutoTokenizer, AutoModel

tokenizer = AutoTokenizer._pretrained("THUDM/codegeex2-6b", trust_remote_code= True)

model = AutoModel._pretrained("THUDM/codegeex2-6b", trust_remote_code= True, device= 'cuda')

model = model.eval

# remember adding a language tag for better performance

prompt ="# language: python\n# write a bubble sort function\n"

inputs = tokenizer.encode(prompt, return_tensors="pt").to(model.device)

outputs = model.generate(inputs, max_length= 256, top_k= 1)

response = tokenizer.decode(outputs[ 0])

>>> print(response)

# language: python

# write a bubble sort function

https://arxiv.org/abs/2303.17568

Github項目頁:

https://github.com/THUDM/CodeGeeX2

Hugging Face項目頁：

https://huggingface.co/THUDM/codegeex2-6b

—完—

「AIGC算力全景與趨勢報告 · 量子位智庫」下載

AIGC時代，算力成為「硬通貨」。當前大模型的發展對算力產生新的需求，算力行業迎來變革機會。機遇當前，量子位智庫發布《AIGC算力全景與趨勢報告》，掃描下方二維碼即可查看完整報告。

8月9日，我們將在北京舉辦「量子位AIGC行業沙龍——AIGC時代的算力基石」，歡迎感興趣的企業聯系～

一鍵三連「分享」、「點贊」和「在看」

科技前沿進展日日相見 ~