Spaces:
Sleeping
Sleeping
| import torch | |
| class Config(object): | |
| def __init__(self) -> None: | |
| self.DEVICE = torch.device("cpu") | |
| self.VOCAB_FILE = 'source/vocabulary.txt' | |
| self.VOCAB_SIZE = 5000 | |
| self.NUM_LAYER = 1 | |
| self.IMAGE_EMB_DIM = 512 | |
| self.WORD_EMB_DIM = 512 | |
| self.HIDDEN_DIM = 512 | |
| self.EMBEDDING_WEIGHT_FILE = 'source/weights/embeddings-32B-512H-1L-e10.pt' | |
| self.ENCODER_WEIGHT_FILE = 'source/weights/encoder-32B-512H-1L-e10.pt' | |
| self.DECODER_WEIGHT_FILE = 'source/weights/decoder-32B-512H-1L-e10.pt' | |