from local

This commit is contained in:
dohe0342 2023-04-11 16:32:38 +09:00
parent 0e13fd3209
commit b285b9052f
2 changed files with 1 additions and 1 deletions

View File

@ -1622,7 +1622,7 @@ def run_adapter(rank, world_size, args, wb=None):
librispeech = LibriSpeechAsrDataModule(args)
model.prompt = model.prompt.to('cuda')
model.prompt = model.prompt.to(device)
'''
if params.hpo: