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' |