Spaces:
Running
on
Zero
Running
on
Zero
NikhilJoson
commited on
Commit
•
7ff5a45
1
Parent(s):
77a2181
Update app.py
Browse files
app.py
CHANGED
@@ -236,7 +236,7 @@ def generate_mask(inp_image, label, threshold):
|
|
236 |
|
237 |
|
238 |
#Setting up Flux (Schnell) Inpainting
|
239 |
-
text_encoder_ = CLIPTextModel.from_pretrained("openai/clip-vit-large-
|
240 |
text_encoder_2_ = T5EncoderModel.from_pretrained("xlabs-ai/xflux_text_encoders", torch_dtype=torch.bfloat16)
|
241 |
|
242 |
inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell",text_encoder=text_encoder_,text_encoder_2=text_encoder_2_, torch_dtype=torch.bfloat16).to(DEVICE)
|
|
|
236 |
|
237 |
|
238 |
#Setting up Flux (Schnell) Inpainting
|
239 |
+
text_encoder_ = CLIPTextModel.from_pretrained("openai/clip-vit-large-patch14", torch_dtype=torch.bfloat16)
|
240 |
text_encoder_2_ = T5EncoderModel.from_pretrained("xlabs-ai/xflux_text_encoders", torch_dtype=torch.bfloat16)
|
241 |
|
242 |
inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell",text_encoder=text_encoder_,text_encoder_2=text_encoder_2_, torch_dtype=torch.bfloat16).to(DEVICE)
|