Update app.py
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ pipe = FluxPipeline.from_pretrained(
|
|
| 29 |
)
|
| 30 |
|
| 31 |
# Enable CPU offloading to stay within the ~16GB RAM limit
|
| 32 |
-
pipe.enable_model_cpu_offload()
|
| 33 |
|
| 34 |
class PromptRequest(BaseModel):
|
| 35 |
prompt: str
|
|
|
|
| 29 |
)
|
| 30 |
|
| 31 |
# Enable CPU offloading to stay within the ~16GB RAM limit
|
| 32 |
+
# pipe.enable_model_cpu_offload() # Disabled because no GPU was found
|
| 33 |
|
| 34 |
class PromptRequest(BaseModel):
|
| 35 |
prompt: str
|