Hugging Face's logo

Spaces:
bixoryai
/
flux-lora-ft
Runtime error

App Files Community
Andybeyond commited on
Commit
8eb6265
·
verified ·
1 Parent(s): 1042bcf

Update the FLUX model name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
6
  import torchvision.transforms as transforms
7
  import os
8
 
9
- FLUX_MODEL_NAME = "black-forest-labs/FLUX"
10
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
  class FluxDataset(Dataset):
 
6
  import torchvision.transforms as transforms
7
  import os
8
 
9
+ FLUX_MODEL_NAME = "black-forest-labs/FLUX.1-dev"
10
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
11
 
12
  class FluxDataset(Dataset):