openfree commited on
Commit
e555cbd
·
verified ·
1 Parent(s): f2f64b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -70,14 +70,13 @@ pipe_controlnet = FluxControlNetPipeline(
70
  tokenizer=pipe.tokenizer,
71
  text_encoder_2=pipe.text_encoder_2,
72
  tokenizer_2=pipe.tokenizer_2,
73
- transformer=pipe.transformer, # 여기서 수정되었습니다.
74
  controlnet=controlnet,
75
  scheduler=pipe.scheduler,
76
- safety_checker=pipe.safety_checker,
77
- feature_extractor=pipe.feature_extractor,
78
  torch_dtype=dtype
79
  ).to(device)
80
 
 
81
  MAX_SEED = 2**32 - 1
82
  MAX_PIXEL_BUDGET = 1024 * 1024
83
 
 
70
  tokenizer=pipe.tokenizer,
71
  text_encoder_2=pipe.text_encoder_2,
72
  tokenizer_2=pipe.tokenizer_2,
73
+ transformer=pipe.transformer, # unet 대신 transformer 사용
74
  controlnet=controlnet,
75
  scheduler=pipe.scheduler,
 
 
76
  torch_dtype=dtype
77
  ).to(device)
78
 
79
+
80
  MAX_SEED = 2**32 - 1
81
  MAX_PIXEL_BUDGET = 1024 * 1024
82