gokaygokay commited on
Commit
5eb4ef7
·
verified ·
1 Parent(s): 082faa8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -75,7 +75,7 @@ else:
75
  device = torch.device('cuda')
76
 
77
  base_model = "black-forest-labs/FLUX.1-dev"
78
- pipe = FluxPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16, token=huggingface_token).to(device)
79
 
80
  # Load and fuse LoRA BEFORE quantizing
81
  print('Loading and fusing lora, please wait...')
 
75
  device = torch.device('cuda')
76
 
77
  base_model = "black-forest-labs/FLUX.1-dev"
78
+ pipe = FluxPipeline.from_single_file("https://huggingface.co/marduk191/Flux.1_collection/resolve/main/flux.1_dev_fp8_fp16t5-marduk191.safetensors", torch_dtype=torch.bfloat16, token=huggingface_token).to(device)
79
 
80
  # Load and fuse LoRA BEFORE quantizing
81
  print('Loading and fusing lora, please wait...')