Spaces:
Runtime error
Runtime error
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)
|