Instructions to use flax-community/code-mt5-base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use flax-community/code-mt5-base with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("flax-community/code-mt5-base") model = AutoModelForSeq2SeqLM.from_pretrained("flax-community/code-mt5-base", device_map="auto") - Notebooks
- Google Colab
- Kaggle
| # To add a new cell, type '# %%' | |
| # To add a new markdown cell, type '# %% [markdown]' | |
| # %% | |
| # from IPython import get_ipython | |
| # %% | |
| # get_ipython().system("ls -l ../logs") | |
| # %% | |
| # get_ipython().system(" cat ../logs/model_big.log") | |
| # %% | |
| path = "code-mt5.log" | |
| losses = [] | |
| steps = [] | |
| eval_steps = [] | |
| eval_losses = [] | |
| eval_accs = [] | |
| learning_rate = [] | |
| with open(path, "r") as filePtr: | |
| for line in filePtr: | |
| # print(line) | |
| toks = line.split() | |
| if toks[0] == "Step...": | |
| if "Learning" in toks: | |
| losses.append(float(toks[4].split(",")[0])) | |
| steps.append(int(toks[1].split("(")[1])) | |
| learning_rate.append(float(toks[-1].split(")")[0])) | |
| if "Acc:" in toks: | |
| eval_steps.append(int(toks[1].split("(")[1])) | |
| eval_losses.append(float(toks[4].split(",")[0])) | |
| eval_accs.append(float(toks[-1].split(")")[0])) | |
| # %% | |
| import matplotlib.pyplot as plt | |
| # %% | |
| # print(losses) | |
| # print(steps) | |
| # %% | |
| print("Steps done: ", len(losses) * 100) | |
| # %% | |
| print("last 30 losses: ", losses[-30:]) | |
| # %% | |
| plt.plot(steps, losses) | |
| plt.show() | |
| # %% | |
| min_loss, at_step = 1e10, None | |
| for step, loss in zip(steps, losses): | |
| if loss < min_loss: | |
| min_loss = loss | |
| at_step = step | |
| print("min loss: {} at step {}".format(min_loss, at_step)) | |
| # %% | |
| print(eval_losses) | |
| # %% | |
| plt.plot(eval_steps, eval_losses) | |
| plt.show() | |
| # %% | |
| print(eval_accs) | |
| # %% | |
| plt.plot(eval_steps, eval_accs) | |
| plt.show() | |
| # %% | |
| plt.plot(steps, learning_rate) | |
| plt.show() | |
| # %% | |