Hugging Face's logo

Spaces:
LPX55
/
FLUX.MF-Lightning-Fast-Upscaler
Running on Zero

App Files Community
3
LPX55 commited on
Commit
f9ef187
·
verified ·
1 Parent(s): 42c14c6

Update app_v5.py

Browse files
Files changed (1) hide show
  1. app_v5.py +1 -2
app_v5.py CHANGED
@@ -68,8 +68,7 @@ pipeline_quant_config = PipelineQuantizationConfig(
68
  pipe = FluxControlNetPipeline.from_pretrained(
69
  "LPX55/FLUX.1M-8step_upscaler-cnet",
70
  quantization_config=pipeline_quant_config,
71
- torch_dtype=torch.float16,
72
- device_map="auto",
73
  )
74
  pipe.to("cuda")
75
 
 
68
  pipe = FluxControlNetPipeline.from_pretrained(
69
  "LPX55/FLUX.1M-8step_upscaler-cnet",
70
  quantization_config=pipeline_quant_config,
71
+ torch_dtype=torch.float16
 
72
  )
73
  pipe.to("cuda")
74