Instructions to use TheRemixer/misc-small-experiment with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use TheRemixer/misc-small-experiment with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("CabalResearch/NoobAI-Flux2VAE-RectifiedFlow-0.3", dtype=torch.bfloat16, device_map="cuda") pipe.load_lora_weights("TheRemixer/misc-small-experiment") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
- Local Apps
- Draw Things
Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
base_model:
|
| 4 |
+
- CabalResearch/NoobAI-Flux2VAE-RectifiedFlow-0.3
|
| 5 |
+
library_name: diffusers
|
| 6 |
+
tags:
|
| 7 |
+
- lora
|
| 8 |
+
---
|
| 9 |
+
# Files:
|
| 10 |
+
## **Loras/Mugen_NoobFlux2RF_Test-000003.safetensors**
|
| 11 |
+
|
| 12 |
+
Produces different results, *normally* worse, trained for [Mugen](https://huggingface.co/CabalResearch/Mugen) with an auxillary training objective from https://arxiv.org/abs/2411.04873.
|
| 13 |
+
|
| 14 |
+
I decode both the ground truth latent and predicited clean latent, and take the first two up_blocks of the VAE when decoding both the ground truth latent and predicited latent, then compare them against each other using L2 loss and add that back onto the regular flow matching loss at a weight of 0.1
|
| 15 |
+
|
| 16 |
+
(So `loss_total = flow matching + 0.1 * latent_perceptual_loss`). I only did this if the timestep was less than 50% (`sigmas < 0.5`).
|
| 17 |
+
|
| 18 |
+
## **Loras/RF-Flux2VAE-Consistency-Test-50-000002.safetensors**
|
| 19 |
+
|
| 20 |
+
trained by generating one forward pass from 20-50 timestep before the target timestep with **no gradients**. Then simulating one euler step to the target timestep and using the resulting latent as the input to the model and training on that. So the model is trained on `clean latent + noise + discretization error + mispredicition error from previous step`
|
| 21 |
+
|
| 22 |
+
My thinking was at inference time the model **doesn't** only receive the `clean latent + gaussian noise` like in training, but also `+ discretization error + mispredicition error from previous step`
|
| 23 |
+
|
| 24 |
+
Works on [Mugen](https://huggingface.co/CabalResearch/Mugen) too, but trained on [NoobAI-Flux2VAE-RectifiedFlow-0.3](https://huggingface.co/CabalResearch/NoobAI-Flux2VAE-RectifiedFlow-0.3)
|
| 25 |
+
|
| 26 |
+
## **Loras/RF-Flux2VAE-Consistency-Test-000002.safetensors**
|
| 27 |
+
|
| 28 |
+
Same as above, with bigger but worse dataset. Use the above one (the one with 50 in the name) generally.
|