Fix example, load weights safely and remove extra whitespace

by PartyParrot - opened Nov 13

←

Nov 13

•

This PR fixes the example:

Import and model path were wrong (there is no BEN directory)
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 Nov 13

PartyParrot changed pull request status to open Nov 13

MaxwellMeyer changed pull request status to merged Nov 13

Prama LLC org Nov 13

We appreciate you taking the time to improve our repo. You are what makes the world go round!

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment