Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 |
-
|
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,
|