multimodalart HF Staff commited on
Commit
cebd669
1 Parent(s): 6824d71

Load models from mage-flow-community

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +6 -6
README.md CHANGED
@@ -32,4 +32,4 @@ Choose **Mage-Flow-Turbo 路 Fast** (4 steps, CFG 1.0) or **Mage-Flow 路 Quality*
32
 
33
  - Paper: [Mage-Flow: An Efficient Native-Resolution Foundation Model for Image Generation and Editing](https://huggingface.co/papers/2607.19064)
34
  - GitHub: [microsoft/Mage](https://github.com/microsoft/Mage)
35
- - Models: [Mage-Flow](https://huggingface.co/microsoft/Mage-Flow), [Mage-Flow-Turbo](https://huggingface.co/microsoft/Mage-Flow-Turbo), [Mage-Flow-Edit](https://huggingface.co/microsoft/Mage-Flow-Edit), [Mage-Flow-Edit-Turbo](https://huggingface.co/microsoft/Mage-Flow-Edit-Turbo)
 
32
 
33
  - Paper: [Mage-Flow: An Efficient Native-Resolution Foundation Model for Image Generation and Editing](https://huggingface.co/papers/2607.19064)
34
  - GitHub: [microsoft/Mage](https://github.com/microsoft/Mage)
35
+ - Models: [Mage-Flow](https://huggingface.co/mage-flow-community/Mage-Flow), [Mage-Flow-Turbo](https://huggingface.co/mage-flow-community/Mage-Flow-Turbo), [Mage-Flow-Edit](https://huggingface.co/mage-flow-community/Mage-Flow-Edit), [Mage-Flow-Edit-Turbo](https://huggingface.co/mage-flow-community/Mage-Flow-Edit-Turbo)
app.py CHANGED
@@ -19,11 +19,11 @@ from mage_flow.pipeline import MageFlowPipeline
19
 
20
  MODEL_VARIANTS = {
21
  "turbo": {
22
- "t2i": "microsoft/Mage-Flow-Turbo", "edit": "microsoft/Mage-Flow-Edit-Turbo",
23
  "t2i_steps": 4, "edit_steps": 4, "cfg": 1.0,
24
  },
25
  "quality": {
26
- "t2i": "microsoft/Mage-Flow", "edit": "microsoft/Mage-Flow-Edit",
27
  "t2i_steps": 20, "edit_steps": 30, "cfg": 5.0,
28
  },
29
  }
@@ -145,10 +145,10 @@ with gr.Blocks(css=CSS) as demo:
145
  "# Mage-Flow\n"
146
  "Efficient Native-Resolution Foundation Model for Image Generation and Editing. "
147
  "Enter a prompt to generate an image, or upload an image to edit it.\n\n"
148
- "Models: [Mage-Flow](https://huggingface.co/microsoft/Mage-Flow), "
149
- "[Mage-Flow-Turbo](https://huggingface.co/microsoft/Mage-Flow-Turbo), "
150
- "[Mage-Flow-Edit](https://huggingface.co/microsoft/Mage-Flow-Edit), "
151
- "[Mage-Flow-Edit-Turbo](https://huggingface.co/microsoft/Mage-Flow-Edit-Turbo) | "
152
  "[Paper](https://huggingface.co/papers/2607.19064) | "
153
  "[GitHub](https://github.com/microsoft/Mage)"
154
  )
 
19
 
20
  MODEL_VARIANTS = {
21
  "turbo": {
22
+ "t2i": "mage-flow-community/Mage-Flow-Turbo", "edit": "mage-flow-community/Mage-Flow-Edit-Turbo",
23
  "t2i_steps": 4, "edit_steps": 4, "cfg": 1.0,
24
  },
25
  "quality": {
26
+ "t2i": "mage-flow-community/Mage-Flow", "edit": "mage-flow-community/Mage-Flow-Edit",
27
  "t2i_steps": 20, "edit_steps": 30, "cfg": 5.0,
28
  },
29
  }
 
145
  "# Mage-Flow\n"
146
  "Efficient Native-Resolution Foundation Model for Image Generation and Editing. "
147
  "Enter a prompt to generate an image, or upload an image to edit it.\n\n"
148
+ "Models: [Mage-Flow](https://huggingface.co/mage-flow-community/Mage-Flow), "
149
+ "[Mage-Flow-Turbo](https://huggingface.co/mage-flow-community/Mage-Flow-Turbo), "
150
+ "[Mage-Flow-Edit](https://huggingface.co/mage-flow-community/Mage-Flow-Edit), "
151
+ "[Mage-Flow-Edit-Turbo](https://huggingface.co/mage-flow-community/Mage-Flow-Edit-Turbo) | "
152
  "[Paper](https://huggingface.co/papers/2607.19064) | "
153
  "[GitHub](https://github.com/microsoft/Mage)"
154
  )