Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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.
|
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...')
|