ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Sesame

Browse source
This commit is contained in:
Daniel Han 2025-05-15 07:29:32 -07:00
commit e332f91812
3 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
pyproject.toml
View file

@ -37,7 +37,7 @@ triton = [
]
huggingface = [
"unsloth_zoo>=2025.5.5",
"unsloth_zoo>=2025.5.6",
"packaging",
"tyro",
"transformers==4.51.3,!=4.47.0",
@ -381,7 +381,7 @@ colab-ampere-torch220 = [
"flash-attn>=2.6.3",
]
colab-new = [
"unsloth_zoo>=2025.5.5",
"unsloth_zoo>=2025.5.6",
"packaging",
"tyro",
"transformers==4.51.3,!=4.47.0",

2
unsloth/models/_utils.py
View file

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
__version__ = "2025.5.3"
__version__ = "2025.5.4"
__all__ = [
"SUPPORTS_BFLOAT16",

4
unsloth/models/mapper.py
View file

@ -817,6 +817,10 @@ __INT_TO_FLOAT_MAPPER = \
"microsoft/Phi-4-mini-reasoning",
"unsloth/phi-4-mini-reasoning-bnb-4bit",
),
"unsloth/csm-1b" : (
"unsloth/csm-1b",
"sesame/csm-1b",
),
}
INT_TO_FLOAT_MAPPER = {}