Update README.md
Browse files
README.md
CHANGED
|
@@ -197,7 +197,7 @@ vram_config = {
|
|
| 197 |
}
|
| 198 |
pipe = MiniMaxH3Pipeline.from_pretrained(
|
| 199 |
torch_dtype=torch.bfloat16,
|
| 200 |
-
|
| 201 |
model_configs=...,
|
| 202 |
processor_config=...,
|
| 203 |
+ vram_limit=0,
|
|
|
|
| 197 |
}
|
| 198 |
pipe = MiniMaxH3Pipeline.from_pretrained(
|
| 199 |
torch_dtype=torch.bfloat16,
|
| 200 |
+
+ device="mps",
|
| 201 |
model_configs=...,
|
| 202 |
processor_config=...,
|
| 203 |
+ vram_limit=0,
|