Erlangshen-UniMC-Zero-Shot

Build error

suolyer commited on Nov 19, 2022

Commit

6d84157

1 Parent(s): 6d11d05

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -39,6 +39,7 @@ import time
 # os.environ["CUDA_VISIBLE_DEVICES"] = '6'
 class UniMCDataset(Dataset):
     def __init__(self, data, yes_token, no_token, tokenizer, args, used_mask=True):
         super().__init__()

 # os.environ["CUDA_VISIBLE_DEVICES"] = '6'
 class UniMCDataset(Dataset):
     def __init__(self, data, yes_token, no_token, tokenizer, args, used_mask=True):
         super().__init__()