flux-labs

Running on Zero

vilarin commited on May 28, 2024

Commit

5e2e591

verified ·

1 Parent(s): 8a23466

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -90,7 +90,7 @@ class ModelWrapper:
         else:
             raise NotImplementedError()
-        noise = noise.to(torch.float16)
         print(f'noise: {noise.dtype}')
         prompt_embed = prompt_embed.to(torch.float32)
         DTYPE = prompt_embed.dtype

         else:
             raise NotImplementedError()
+        #noise = noise.to(torch.float16)
         print(f'noise: {noise.dtype}')
         prompt_embed = prompt_embed.to(torch.float32)
         DTYPE = prompt_embed.dtype