AlekseyCalvin commited on
Commit
0c99ba1
·
verified ·
1 Parent(s): 6a6110d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ with open('loras.json', 'r') as f:
38
 
39
  dtype = torch.bfloat16
40
 
41
- model = FluxTransformer2DModel.from_pretrained("ostris/OpenFLUX.1", torch_dtype=dtype).to("cuda")
42
  model.num_single_layers="0"
43
  model.chunk_size="0"
44
  model.pooled_projections="(_, _, 1)[0]"
 
38
 
39
  dtype = torch.bfloat16
40
 
41
+ model = FluxTransformer2DModel.from_pretrained("ostris/OpenFLUX.1", subfolder="transformer", torch_dtype=dtype).to("cuda")
42
  model.num_single_layers="0"
43
  model.chunk_size="0"
44
  model.pooled_projections="(_, _, 1)[0]"