THUDM

ChatGLM, CogVLM, CodeGeeX, WebGLM, GLM-130B, CogView, CogVideo | CogDL, GNNs, AMiner | Knowledge Engineering Group (KEG) & Data Mining at Tsinghua University

FIT Building, Tsinghua University

https://huggingface.co/THUDM

THUDM's repositories

GLM-130B

GLM-130B: An Open Bilingual Pre-Trained Model (ICLR 2023)

Language:PythonApache-2.07634 98 198

CogVideo

Text-to-video generation. The repo for ICLR2023 paper "CogVideo: Large-scale Pretraining for Text-to-Video Generation via Transformers"

Language:PythonApache-2.03528 104 38

GLM

GLM (General Language Model)

Language:PythonMIT3074 46 190

P-tuning-v2

An optimized deep prompt tuning strategy comparable to fine-tuning across scales and tasks

Language:PythonApache-2.01915 30 72

CogView

Text-to-Image generation. The repo for NeurIPS 2021 paper "CogView: Mastering Text-to-Image Generation via Transformers".

Language:PythonApache-2.01623 56 63

WebGLM

WebGLM: An Efficient Web-enhanced Question Answering System (KDD 2023)

Language:PythonApache-2.01524 25 69

AgentTuning

AgentTuning: Enabling Generalized Agent Abilities for LLMs

Language:Python1256 16 47

ImageReward

[NeurIPS 2023] ImageReward: Learning and Evaluating Human Preferences for Text-to-image Generation

Language:PythonApache-2.0992 14 74

CogQA

Source code and dataset for ACL 2019 paper "Cognitive Graph for Multi-Hop Reading Comprehension at Scale"

Language:PythonMIT455 19 39

GraphMAE

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

Language:Python434 13 62

GCC

GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training @ KDD 2020

Language:PythonMIT320 15 25

HGB

Revisiting, benchmarking, and refining Heterogeneous Graph Neural Networks.

Language:Python291 7 60

KOBE

Towards Knowledge-Based Personalized Product Description Generation in E-commerce @ KDD 2019

Language:PythonMIT238 8 52

Chinese-Transformer-XL

Language:Python214 10 19

GRAND

Source code and dataset of the NeurIPS 2020 paper "Graph Random Neural Network for Semi-Supervised Learning on Graphs"

Language:PythonMIT202 5 13

icetk

A unified tokenization tool for Images, Chinese and English.

Language:Python145 10 10

GraphMAE2

GraphMAE2: A Decoding-Enhanced Masked Self-Supervised Graph Learner in WWW'23

Language:PythonMIT123 10 8

iPrompt

Code, Data and Demo for Paper: Controllable Generation from Pre-trained Language Models via Inverse Prompting

Language:Python120 12 8

grb

Graph Robustness Benchmark: A scalable, unified, modular, and reproducible benchmark for evaluating the adversarial robustness of Graph Machine Learning.

Language:PythonMIT90 8 10

CogKR

Source code and dataset for paper "Cognitive Knowledge Graph Reasoning for One-shot Relational Learning"

Language:Python71 6 2

XDAI

Language:Python62 10 4

FasterTransformer

Transformer related optimization, including BERT, GPT

Language:C++Apache-2.040 30

FastLDM

Inference speed-up for stable-diffusion (ldm) with TensorRT.

Language:PythonApache-2.034 4 3

ApeGNN

Language:PythonMIT19 100

BatchSampler

The source code for BatchSampler that accepted in KDD'23

Language:Python17 10 1

MRT

MRT: Tracing the Evolution of Scientific Publications (TKDE 2021)

WinGNN

Language:Python6 10

DropConn

Language:Python5 90

DistAlign-GNNs

Language:Jupyter NotebookMIT3 110

open_clip_pix2struct

pix2struct version of open_clip

Language:Jupyter NotebookNOASSERTION3 7 1