MohamedRashad commited on
Commit
09d1730
·
1 Parent(s): e8fcedb

Remove redundant torch import and streamline device initialization in app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,7 @@
1
  from diffusers import FluxPipeline, AutoencoderKL
2
  from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_images
 
 
3
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
4
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to(device)
5
  good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
@@ -12,7 +14,6 @@ from gradio_litmodel3d import LitModel3D
12
  import os
13
  os.environ['SPCONV_ALGO'] = 'native'
14
  from typing import *
15
- import torch
16
  import numpy as np
17
  import imageio
18
  import uuid
 
1
  from diffusers import FluxPipeline, AutoencoderKL
2
  from live_preview_helpers import flux_pipe_call_that_returns_an_iterable_of_images
3
+ import torch
4
+
5
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
6
  pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to(device)
7
  good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16).to(device)
 
14
  import os
15
  os.environ['SPCONV_ALGO'] = 'native'
16
  from typing import *
 
17
  import numpy as np
18
  import imageio
19
  import uuid