just testing cublas

This commit is contained in:
Concedo 2023-05-15 20:01:22 +08:00
parent fce2e7e518
commit 6504150fac
9 changed files with 53 additions and 59 deletions

View file

@ -10,7 +10,7 @@ import torch
import numpy as np
import re
from transformers import GPTJForCausalLM, AutoModelForCausalLM
from transformers import AutoModelForCausalLM
# ref: https://github.com/openai/gpt-2/blob/master/src/encoder.py
def bytes_to_unicode():