• 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
.github Loading commit data...
embeddings Loading commit data...
extensions Loading commit data...
extensions-builtin Loading commit data...
javascript Loading commit data...
localizations Loading commit data...
models Loading commit data...
modules Loading commit data...
scripts Loading commit data...
test Loading commit data...
textual_inversion_templates Loading commit data...
.gitignore Loading commit data...
.pylintrc Loading commit data...
CODEOWNERS Loading commit data...
README.md Loading commit data...
artists.csv Loading commit data...
environment-wsl2.yaml Loading commit data...
launch.py Loading commit data...
requirements.txt Loading commit data...
requirements_versions.txt Loading commit data...
screenshot.png Loading commit data...
script.js Loading commit data...
style.css Loading commit data...
txt2img_Screenshot.png Loading commit data...
v1-inference.yaml Loading commit data...
webui-macos-env.sh Loading commit data...
webui-user.bat Loading commit data...
webui-user.sh Loading commit data...
webui.bat Loading commit data...
webui.py Loading commit data...
webui.sh Loading commit data...