AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

guard accelerate import.

Browse Source
This commit is contained in:
sayakpaul
2026-01-07 09:50:34 +05:30
parent 2f35e145e3
commit 1b98e10614
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/diffusers/training_utils.py
View File

@@ -11,7 +11,6 @@ from typing import Any, Dict, Iterable, List, Optional, Set, Tuple, Type, Union
import numpy as np
import torch
from accelerate.logging import get_logger
if getattr(torch, "distributed", None) is not None:
@@ -26,6 +25,7 @@ from .utils import (
convert_state_dict_to_diffusers,
convert_state_dict_to_peft,
deprecate,
is_accelerate_available,
is_peft_available,
is_torch_npu_available,
is_torchvision_available,
@@ -39,6 +39,9 @@ if is_transformers_available():
if transformers.integrations.deepspeed.is_deepspeed_zero3_enabled():
import deepspeed
if is_accelerate_available():
from accelerate.logging import get_logger
if is_peft_available():
from peft import set_peft_model_state_dict