openfree commited on
Commit
81269c1
·
verified ·
1 Parent(s): 267d71c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -63,12 +63,13 @@ controlnet = FluxControlNetModel.from_pretrained(
63
  "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
64
  )
65
 
 
66
  # Upscale 파이프라인 설정 (기존 pipe 재사용)
67
  pipe_upscale = FluxControlNetPipeline(
68
  vae=pipe.vae,
69
  text_encoder=pipe.text_encoder,
70
  tokenizer=pipe.tokenizer,
71
- unet=pipe.unet,
72
  scheduler=pipe.scheduler,
73
  safety_checker=pipe.safety_checker,
74
  feature_extractor=pipe.feature_extractor,
 
63
  "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
64
  )
65
 
66
+ # Upscale 파이프라인 설정 (기존 pipe 재사용)
67
  # Upscale 파이프라인 설정 (기존 pipe 재사용)
68
  pipe_upscale = FluxControlNetPipeline(
69
  vae=pipe.vae,
70
  text_encoder=pipe.text_encoder,
71
  tokenizer=pipe.tokenizer,
72
+ transformer=pipe.transformer, # FLUX 모델은 transformer를 사용할 수 있습니다
73
  scheduler=pipe.scheduler,
74
  safety_checker=pipe.safety_checker,
75
  feature_extractor=pipe.feature_extractor,