• brkirch's avatar
    Add numpy fix for MPS on PyTorch 1.12.1 · 16b4509f
    brkirch authored
    When saving training results with torch.save(), an exception is thrown:
    "RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead."
    
    So for MPS, check if Tensor.requires_grad and detach() if necessary.
    16b4509f
Name
Last commit
Last update
..
api Loading commit data...
codeformer Loading commit data...
hypernetworks Loading commit data...
textual_inversion Loading commit data...
artists.py Loading commit data...
call_queue.py Loading commit data...
codeformer_model.py Loading commit data...
deepbooru.py Loading commit data...
deepbooru_model.py Loading commit data...
devices.py Loading commit data...
errors.py Loading commit data...
esrgan_model.py Loading commit data...
esrgan_model_arch.py Loading commit data...
extensions.py Loading commit data...
extras.py Loading commit data...
face_restoration.py Loading commit data...
generation_parameters_copypaste.py Loading commit data...
gfpgan_model.py Loading commit data...
images.py Loading commit data...
img2img.py Loading commit data...
interrogate.py Loading commit data...
localization.py Loading commit data...
lowvram.py Loading commit data...
masking.py Loading commit data...
memmon.py Loading commit data...
modelloader.py Loading commit data...
ngrok.py Loading commit data...
paths.py Loading commit data...
processing.py Loading commit data...
prompt_parser.py Loading commit data...
realesrgan_model.py Loading commit data...
safe.py Loading commit data...
script_callbacks.py Loading commit data...
script_loading.py Loading commit data...
scripts.py Loading commit data...
sd_hijack.py Loading commit data...
sd_hijack_checkpoint.py Loading commit data...
sd_hijack_clip.py Loading commit data...
sd_hijack_inpainting.py Loading commit data...
sd_hijack_open_clip.py Loading commit data...
sd_hijack_optimizations.py Loading commit data...
sd_hijack_unet.py Loading commit data...
sd_models.py Loading commit data...
sd_samplers.py Loading commit data...
sd_vae.py Loading commit data...
shared.py Loading commit data...
styles.py Loading commit data...
txt2img.py Loading commit data...
ui.py Loading commit data...
ui_extensions.py Loading commit data...
ui_tempdir.py Loading commit data...
upscaler.py Loading commit data...