Skip to content

Commit

Permalink
Update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
DarkMnDragon committed Dec 15, 2024
1 parent b1faf8e commit e54c261
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,15 @@ return self.criterion(

```

After training, converting a pretrained rectified flow to another interpolation scheme (as long as alpha and beta are specified) can be done easily and automatically by:
```python
from rectified_flow.flow_components.interpolation_convertor import AffineInterpConverter

# Converting pretrained rf into spherical one
target_interp = AffineInterp("spherical")
converted_spherical_rf = AffineInterpConverter(rf, target_interp).transform_rectified_flow()
```

For sampling, import the desired sampler class and pass the `RectifiedFlow` instance to it.
```python
from rectified_flow.samplers import SDESampler
Expand Down

0 comments on commit e54c261

Please sign in to comment.