Muggle666

The LLAMA model is from https://github.com/juncongmoo/pyllama/tree/main. And I just try to add lora on it. The full model code is: ``` from typing import Optional, Tuple from dataclasses import dataclass...

``` class CustomDataset(Dataset): def __init__(self, data, labels): self.data = data self.labels = labels def __len__(self): return len(self.data) def __getitem__(self, index): sample = self.data[index] label = self.labels[index] return torch.tensor(sample, dtype=torch.long), torch.tensor(label,...