File size: 260 Bytes
f3f8c30 |
1 2 3 4 5 6 7 |
from transformers import PretrainedConfig
import torch
class DeepFakeConfig(PretrainedConfig):
model_type = "ResNet"
def __init__(self,**kwargs):
super().__init__(**kwargs)
self.DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu' |