wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-07-31 12:00:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Apply pre-commit

Browse Source
This commit is contained in:
hyunwoongko
2023-02-08 03:50:27 +09:00
parent 44c555cad1
commit cb722768f7
3 changed files with 100 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model/supervised_finetuning/efficiency_utils.py
+1 -1
View File
@@ -1,7 +1,7 @@
import functools
import torch
from transformers.activations import QuickGELUActivation, NewGELUActivation, FastGELUActivation, GELUActivation
from transformers.activations import FastGELUActivation, GELUActivation, NewGELUActivation, QuickGELUActivation
def rsetattr(obj, attr, val):
model/supervised_finetuning/trainer.py
+1 -1
View File
@@ -5,11 +5,11 @@ from typing import Any, Dict, List, Optional, Tuple, Union
import bitsandbytes
import torch
from efficiency_utils import fuse_gelu
from torch import nn
from transformers import PreTrainedModel, Trainer, TrainingArguments
from transformers.training_args import OptimizerNames
from utils import get_dataset, get_loss, get_metrics, get_model, get_tokenizer, read_yamls
from efficiency_utils import fuse_gelu
def compute_metrics(eval_pred, preprocess_fns, metrics):