bloom

9fdb7dab · yuguo960516 · 9fdb7dab · 9fdb7dab · 9fdb7dab · 9fdb7dab
Commit 9fdb7dab authored Mar 30, 2023 by yuguo960516
20 changed files
--- a/libai/inference/text_generation.py
+++ b/libai/inference/text_generation.py
--- a/libai/inference/utils/imagenet_class.py
+++ b/libai/inference/utils/imagenet_class.py
--- a/libai/layers/__init__.py
+++ b/libai/layers/__init__.py
--- a/libai/layers/activation.py
+++ b/libai/layers/activation.py
--- a/libai/layers/attention.py
+++ b/libai/layers/attention.py
--- a/libai/layers/conv.py
+++ b/libai/layers/conv.py
--- a/libai/layers/cross_entropy.py
+++ b/libai/layers/cross_entropy.py
--- a/libai/layers/droppath.py
+++ b/libai/layers/droppath.py
--- a/libai/layers/embedding.py
+++ b/libai/layers/embedding.py
--- a/libai/layers/layer_norm.py
+++ b/libai/layers/layer_norm.py
--- a/libai/layers/linear.py
+++ b/libai/layers/linear.py
--- a/libai/layers/lm_logits.py
+++ b/libai/layers/lm_logits.py
--- a/libai/layers/mlp.py
+++ b/libai/layers/mlp.py
--- a/libai/layers/transformer_layer.py
+++ b/libai/layers/transformer_layer.py
--- a/libai/models/__init__.py
+++ b/libai/models/__init__.py
--- a/libai/models/bert_model.py
+++ b/libai/models/bert_model.py
--- a/libai/models/build.py
+++ b/libai/models/build.py
--- a/libai/models/gpt_model.py
+++ b/libai/models/gpt_model.py
--- a/libai/models/resmlp.py
+++ b/libai/models/resmlp.py
--- a/libai/models/roberta_model.py
+++ b/libai/models/roberta_model.py