Wi-zz

/

joy-caption-pre-alpha

like 57

Could you please try the app-multi-alpha.py file in the main repo to see if multi-gpu is working for you. Usage is the same.

There's a bug in the multi script. Line 81

with torch.amp.autocast_mode.autocast(rank, enabled=True)

Should be:

with torch.amp.autocast_mode.autocast(device_type='cuda', enabled=True):

Otherwise it tries to assign the integer of rank to device_type which expects a string.

There's a bug in the multi script. Line 81

with torch.amp.autocast_mode.autocast(rank, enabled=True)

Should be:

with torch.amp.autocast_mode.autocast(device_type='cuda', enabled=True):

Otherwise it tries to assign the integer of rank to device_type which expects a string.

Thank you, I've updated the script.