Jitesh Dhamaniya commited on
Commit ·
3890bcb
1
Parent(s): 5ec0f36
update model id
Browse files- handler.py +2 -2
- main.py +1 -1
handler.py
CHANGED
|
@@ -6,8 +6,8 @@ from io import BytesIO
|
|
| 6 |
import base64
|
| 7 |
|
| 8 |
# Load the model once (caching for efficiency)
|
| 9 |
-
MODEL_ID = "alimama-creative
|
| 10 |
-
CONTROLNET_MODEL = "alimama-creative
|
| 11 |
TRANSFORMER_MODEL = "black-forest-labs/FLUX.1-dev"
|
| 12 |
|
| 13 |
controlnet = DiffusionPipeline.from_pretrained(CONTROLNET_MODEL, torch_dtype=torch.bfloat16)
|
|
|
|
| 6 |
import base64
|
| 7 |
|
| 8 |
# Load the model once (caching for efficiency)
|
| 9 |
+
MODEL_ID = "jiteshdhamaniya/alimama-creative-FLUX.1-dev-Controlnet-Inpainting-Alpha"
|
| 10 |
+
CONTROLNET_MODEL = "jiteshdhamaniya/alimama-creative-FLUX.1-dev-Controlnet-Inpainting-Alpha"
|
| 11 |
TRANSFORMER_MODEL = "black-forest-labs/FLUX.1-dev"
|
| 12 |
|
| 13 |
controlnet = DiffusionPipeline.from_pretrained(CONTROLNET_MODEL, torch_dtype=torch.bfloat16)
|
main.py
CHANGED
|
@@ -21,7 +21,7 @@ prompt='a person wearing a white shoe, carrying a white bucket with text "Jitesh
|
|
| 21 |
# print(f"Checkpoint file: {checkpoint_file}")
|
| 22 |
|
| 23 |
# Build pipeline
|
| 24 |
-
controlnet = FluxControlNetModel.from_pretrained("alimama-creative
|
| 25 |
transformer = FluxTransformer2DModel.from_pretrained(
|
| 26 |
"black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16, local_files_only=True
|
| 27 |
)
|
|
|
|
| 21 |
# print(f"Checkpoint file: {checkpoint_file}")
|
| 22 |
|
| 23 |
# Build pipeline
|
| 24 |
+
controlnet = FluxControlNetModel.from_pretrained("jiteshdhamaniya/alimama-creative-FLUX.1-dev-Controlnet-Inpainting-Alpha", torch_dtype=torch.bfloat16, local_files_only=True)
|
| 25 |
transformer = FluxTransformer2DModel.from_pretrained(
|
| 26 |
"black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16, local_files_only=True
|
| 27 |
)
|