File size: 260 Bytes
7c1c90f
 
 
 
 
 
2374ae7
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'