Add utils and reuse functions in torch and flax pipelines

nateraw / stable-diffusion-videos

Create 🔥 videos with Stable Diffusion by exploring the latent space and morphing between text prompts

Apache License 2.0

4.34k stars 413 forks source link

Add utils and reuse functions in torch and flax pipelines #157

Closed charlielito closed 1 year ago

charlielito commented 1 year ago

Added utils.py to reuse some functions in both pipelines (flax and torch)

charlielito commented 1 year ago

Just added the missing features for flax:

Implemented negative_prompt in the flax pipeline
Implemented image_generation for Flax to be used in the Gradio App

nateraw commented 1 year ago

Thanks again for a great contribution!! Read through the code in depth last night. Looks good I think - I'll get this approved and merged this week sometime...got a few projects I'm juggling rn.

This PR really reminds me I should close out #54 so we dont run into diffs with style changes applied, makes PR changes a bit harder to read. Not your fault, my fault.

Unrelated: By the way @charlielito - do you have a Twitter? I'd like to tweet about the TPU release + tag you if you have one.

charlielito commented 1 year ago

This PR really reminds me I should close out #54 so we dont run into diffs with style changes applied, makes PR changes a bit harder to read. Not your fault, my fault.

Yeah, I know :( I tried to configure black (number of chars per line) to not change that much the already existing code.

Unrelated: By the way @charlielito - do you have a Twitter? I'd like to tweet about the TPU release + tag you if you have one.

For sure! my user is @charlielito42