Small fixes
This commit is contained in:
parent
b7e96dde11
commit
10cf55040a
4 changed files with 5 additions and 5 deletions
|
|
@ -37,7 +37,7 @@ triton = [
|
|||
]
|
||||
|
||||
huggingface = [
|
||||
"unsloth_zoo>=2025.7.1",
|
||||
"unsloth_zoo>=2025.7.2",
|
||||
"packaging",
|
||||
"tyro",
|
||||
"transformers>=4.51.3,!=4.47.0,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3",
|
||||
|
|
@ -381,7 +381,7 @@ colab-ampere-torch220 = [
|
|||
"flash-attn>=2.6.3",
|
||||
]
|
||||
colab-new = [
|
||||
"unsloth_zoo>=2025.7.1",
|
||||
"unsloth_zoo>=2025.7.2",
|
||||
"packaging",
|
||||
"tyro",
|
||||
"transformers>=4.51.3,!=4.47.0,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3",
|
||||
|
|
|
|||
|
|
@ -79,7 +79,7 @@ def get_device_count():
|
|||
elif DEVICE_TYPE == "xpu":
|
||||
return torch.xpu.device_count()
|
||||
else:
|
||||
return 0
|
||||
return 1
|
||||
pass
|
||||
|
||||
DEVICE_COUNT : int = get_device_count()
|
||||
|
|
|
|||
|
|
@ -12,7 +12,7 @@
|
|||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
__version__ = "2025.7.1"
|
||||
__version__ = "2025.7.2"
|
||||
|
||||
__all__ = [
|
||||
"SUPPORTS_BFLOAT16",
|
||||
|
|
|
|||
|
|
@ -796,7 +796,7 @@ def LlamaModel_fast_forward(
|
|||
# Ignore attention_mask
|
||||
if attention_mask is None:
|
||||
padding_mask = None
|
||||
elif self.training and os.environ.get("UNSLOTH_KEEP_PADDING", "0") != '1':
|
||||
elif self.training:
|
||||
attention_mask = None
|
||||
padding_mask = None
|
||||
else:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue