Hugging Face's logo

Spaces:
LPX55
/
FLUX.1-Redux_Turbo
Runtime error

App Files Community
LPX55 commited on
Commit
94b53ce
·
verified ·
1 Parent(s): 7e5d218

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -15,8 +15,9 @@ MAX_SEED = np.iinfo(np.int32).max
15
  MAX_IMAGE_SIZE = 2048
16
 
17
  pipe = FluxPipeline.from_pretrained(
18
- "LPX55/FLUX.1-merged_uncensored",
19
- torch_dtype=torch.bfloat16
 
20
  ).to("cuda")
21
  pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
22
  pipe.fuse_lora(lora_scale=0.125)
 
15
  MAX_IMAGE_SIZE = 2048
16
 
17
  pipe = FluxPipeline.from_pretrained(
18
+ "black-forest-labs/FLUX.1-dev",
19
+ torch_dtype=torch.bfloat16,
20
+ token=os.getenv('HF_TOKEN'),
21
  ).to("cuda")
22
  pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"), lora_scale=0.125)
23
  pipe.fuse_lora(lora_scale=0.125)