Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
KhaiQuang
/
ApecGPT-LoRA-v1
like
0
PEFT
TensorBoard
Safetensors
mpt
trl
sft
Generated from Trainer
custom_code
Model card
Files
Files and versions
Metrics
Training metrics
Community
Use this model
84f4212
ApecGPT-LoRA-v1
/
fc.py
KhaiQuang
Upload 16 files
84f4212
verified
5 months ago
raw
Copy download link
history
blame
No virus
167 Bytes
from
torch
import
nn
FC_CLASS_REGISTRY = {
'torch'
: nn.Linear}
try
:
import
transformer_engine.pytorch
as
te
FC_CLASS_REGISTRY[
'te'
] = te.Linear
except
:
pass