Hatman commited on
Commit
f18862a
·
verified ·
1 Parent(s): 3dd4f3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -16,9 +16,6 @@ model_path = 'stabilityai/stable-diffusion-3.5-large'
16
  ip_adapter_path = './ip-adapter.bin'
17
  image_encoder_path = "google/siglip-so400m-patch14-384"
18
 
19
- device = "cuda" if torch.cuda.is_available() else "cpu"
20
- dtype = torch.float16 if torch.cuda.is_available() else torch.float32
21
-
22
  transformer = SD3Transformer2DModel.from_pretrained(
23
  model_path, subfolder="transformer", torch_dtype=torch.bfloat16
24
  )
@@ -33,8 +30,6 @@ pipe.init_ipadapter(
33
  nb_token=64,
34
  )
35
 
36
- pipe.to(device)
37
-
38
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
39
  if randomize_seed:
40
  seed = random.randint(0, 2000)
 
16
  ip_adapter_path = './ip-adapter.bin'
17
  image_encoder_path = "google/siglip-so400m-patch14-384"
18
 
 
 
 
19
  transformer = SD3Transformer2DModel.from_pretrained(
20
  model_path, subfolder="transformer", torch_dtype=torch.bfloat16
21
  )
 
30
  nb_token=64,
31
  )
32
 
 
 
33
  def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
34
  if randomize_seed:
35
  seed = random.randint(0, 2000)