AlekseyCalvin commited on
Commit
758ccac
1 Parent(s): 6efdd95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
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("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
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)