Fix example, load weights safely and remove extra whitespace
#2
by
PartyParrot
- opened
This PR fixes the example:
- Import and model path were wrong (there is no
BENdirectory) - Use inference mode so this actually runs on 6 GB GPUs (there is no need to compute gradients for inference)
- Load weights without opportunity for arbitrary code execution backdoor (
torch.load(..., weights_only=True)) - Add missing image2 (I cropped the windmill from the demo image)
PartyParrot
changed pull request title from
fix-example
to Fix example, load weights safely and remove extra whitespace
PartyParrot
changed pull request status to
open
PramaLLC
changed pull request status to
merged
We appreciate you taking the time to improve our repo. You are what makes the world go round!