Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,6 +39,7 @@ import time
|
|
| 39 |
# os.environ["CUDA_VISIBLE_DEVICES"] = '6'
|
| 40 |
|
| 41 |
|
|
|
|
| 42 |
class UniMCDataset(Dataset):
|
| 43 |
def __init__(self, data, yes_token, no_token, tokenizer, args, used_mask=True):
|
| 44 |
super().__init__()
|
|
|
|
| 39 |
# os.environ["CUDA_VISIBLE_DEVICES"] = '6'
|
| 40 |
|
| 41 |
|
| 42 |
+
|
| 43 |
class UniMCDataset(Dataset):
|
| 44 |
def __init__(self, data, yes_token, no_token, tokenizer, args, used_mask=True):
|
| 45 |
super().__init__()
|