search

sshh12 / terrain-diffusion

MIT License
17 stars 2 forks source link

Inference code #5

Open fatemehtd opened 4 months ago

fatemehtd commented 4 months ago

I was wondering how do you load the weights after fine-tuning? Is your repo containing that part?

sshh12 commented 4 months ago

Hey! it's in the app code here https://github.com/sshh12/terrain-diffusion-app/blob/d8e011cc3aa65667cbe613204e247af33946d6e2/modal/modal_app.py#L63