small fix and update readme

README.md

@@ -133,6 +133,5 @@ Luxury food photograph of an italian Linguine pasta alle vongole dish with lots
 
 ### TODO
 
-- FLUX Dev implementation
 - LoRA adapters
 - Command line args

src/flux_1_schnell/flux.py

@@ -33,7 +33,7 @@ class Flux1:
     def generate_image(self, seed: int, prompt: str, config: Config = Config()) -> PIL.Image.Image:
         sigmas = get_sigmas(config.num_inference_steps)
         if self.is_dev:
-            sigmas = shift_sigmas(sigmas)
+            sigmas = shift_sigmas(sigmas, config.width, config.height)
         latents = LatentCreator.create(config.height, config.width, seed)
 
         t5_tokens = self.t5_tokenizer.tokenize(prompt)