Spaces:
Running
on
Zero
Running
on
Zero
AlekseyCalvin
commited on
Commit
•
758ccac
1
Parent(s):
6efdd95
Update app.py
Browse files
app.py
CHANGED
@@ -24,13 +24,9 @@ os.environ["HF_HOME"] = cache_path
|
|
24 |
|
25 |
torch.backends.cuda.matmul.allow_tf32 = True
|
26 |
|
27 |
-
pipe = FluxPipeline.from_pretrained("
|
28 |
-
pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
|
29 |
-
pipe.fuse_lora(lora_scale=0.125)
|
30 |
pipe.to(device="cuda", dtype=torch.bfloat16)
|
31 |
|
32 |
-
pipe.enable_model_cpu_offload()
|
33 |
-
|
34 |
# Load LoRAs from JSON file
|
35 |
with open('loras.json', 'r') as f:
|
36 |
loras = json.load(f)
|
|
|
24 |
|
25 |
torch.backends.cuda.matmul.allow_tf32 = True
|
26 |
|
27 |
+
pipe = FluxPipeline.from_pretrained("John6666/hyper-flux1-dev-fp8-flux", torch_dtype=torch.bfloat16)
|
|
|
|
|
28 |
pipe.to(device="cuda", dtype=torch.bfloat16)
|
29 |
|
|
|
|
|
30 |
# Load LoRAs from JSON file
|
31 |
with open('loras.json', 'r') as f:
|
32 |
loras = json.load(f)
|