Update README.md
Browse files
README.md
CHANGED
|
@@ -61,6 +61,16 @@ import os, re
|
|
| 61 |
import torch
|
| 62 |
torch._dynamo.config.recompile_limit = 64
|
| 63 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 64 |
#------------
|
| 65 |
|
| 66 |
from unsloth import FastModel
|
|
|
|
| 61 |
import torch
|
| 62 |
torch._dynamo.config.recompile_limit = 64
|
| 63 |
|
| 64 |
+
|
| 65 |
+
import warnings
|
| 66 |
+
|
| 67 |
+
# Suppress the specific PyTorch size check warning from bitsandbytes
|
| 68 |
+
warnings.filterwarnings(
|
| 69 |
+
"ignore",
|
| 70 |
+
category=FutureWarning,
|
| 71 |
+
message=".*_check_is_size will be removed in a future PyTorch release.*"
|
| 72 |
+
)
|
| 73 |
+
|
| 74 |
#------------
|
| 75 |
|
| 76 |
from unsloth import FastModel
|