rafaaa2105 commited on
Commit
23071ef
1 Parent(s): fb3ffbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ if torch.cuda.is_available():
13
  else:
14
  torch_dtype = torch.float32
15
 
16
- pipe = pipeline = AutoPipelineForText2Image.from_pretrained("ostris/OpenFLUX.1", torch_dtype=torch.bfloat16)
17
  pipe.load_lora_weights('aleksa-codes/flux-ghibsky-illustration', weight_name='lora.safetensors')
18
  pipe = pipe.to(device)
19
 
 
13
  else:
14
  torch_dtype = torch.float32
15
 
16
+ pipe = pipeline = AutoPipelineForText2Image.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
17
  pipe.load_lora_weights('aleksa-codes/flux-ghibsky-illustration', weight_name='lora.safetensors')
18
  pipe = pipe.to(device)
19