Hugging Face's logo

Spaces:
LPX55
/
FluxM-Lightning-Upscaler
Running on Zero

App Files Community
2
LPX55 commited on
Commit
d939c81
·
verified ·
1 Parent(s): e5d1cff

Update raw.py

Browse files
Files changed (1) hide show
  1. raw.py +1 -1
raw.py CHANGED
@@ -6,7 +6,7 @@ from diffusers import FluxControlNetModel, FluxControlNetPipeline, AutoencoderKL
6
  import gradio as gr
7
  huggingface_token = os.getenv("HUGGINFACE_TOKEN")
8
 
9
- good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16, token=huggingface_token).to(device)
10
 
11
  # Load pipeline
12
  controlnet = FluxControlNetModel.from_pretrained(
 
6
  import gradio as gr
7
  huggingface_token = os.getenv("HUGGINFACE_TOKEN")
8
 
9
+ good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=torch.bfloat16, token=huggingface_token).to("cuda")
10
 
11
  # Load pipeline
12
  controlnet = FluxControlNetModel.from_pretrained(