amos1088 commited on
Commit
67c0ca0
·
1 Parent(s): e129330
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,11 +34,11 @@ ip_adapter_path = './ip-adapter.bin'
34
  image_encoder_path = "google/siglip-so400m-patch14-384"
35
 
36
  transformer = SD3Transformer2DModel.from_pretrained(
37
- model_path, subfolder="transformer", torch_dtype=torch.bfloat16
38
  )
39
 
40
  pipe = StableDiffusion3Pipeline.from_pretrained(
41
- model_path, transformer=transformer, torch_dtype=torch.bfloat16
42
  ).to("cuda")
43
 
44
  pipe.init_ipadapter(
 
34
  image_encoder_path = "google/siglip-so400m-patch14-384"
35
 
36
  transformer = SD3Transformer2DModel.from_pretrained(
37
+ model_path, subfolder="transformer", torch_dtype=torch.float16
38
  )
39
 
40
  pipe = StableDiffusion3Pipeline.from_pretrained(
41
+ model_path, transformer=transformer, torch_dtype=torch.float16
42
  ).to("cuda")
43
 
44
  pipe.init_ipadapter(