mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2025-09-16 20:09:41 +00:00
just testing cublas
This commit is contained in:
parent
fce2e7e518
commit
6504150fac
9 changed files with 53 additions and 59 deletions
|
@ -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():
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue