Skip to content

Colab not working #3486

@lynrayy

Description

@lynrayy

Issue when launch training

15:42:37-021177 INFO     accelerate launch --num_cpu_threads_per_process=2      
                         "./train_network.py" --v2                              
                         --pretrained_model_name_or_path="/content/drive/MyDrive
                         /AI/models/indigoFurryMix_v120Hybrid.safetensors"      
                         --train_data_dir="/content/drive/MyDrive/AI/dataset"   
                         --resolution="512,512"                                 
                         --output_dir="/content/drive/MyDrive/AI/loras/mvmw"    
                         --network_alpha="144" --save_model_as=safetensors      
                         --network_module=networks.lora --text_encoder_lr=5e-05 
                         --unet_lr=0.0001 --network_dim=144                     
                         --output_name="lora_mvmw" --lr_scheduler_num_cycles="1"
                         --no_half_vae --learning_rate="0.0001"                 
                         --lr_scheduler="constant" --train_batch_size="1"       
                         --max_train_steps="372" --save_every_n_epochs="1"      
                         --mixed_precision="bf16" --save_precision="bf16"       
                         --cache_latents --optimizer_type="AdamW8bit"           
                         --max_data_loader_n_workers="0" --bucket_reso_steps=64 
                         --xformers --bucket_no_upscale --sample_sampler=euler_a
                         --sample_prompts="/content/drive/MyDrive/AI/loras/mvmw/
                         sample/prompt.txt" --sample_every_n_epochs="5"         
Traceback (most recent call last):
  File "/usr/local/bin/accelerate", line 5, in <module>
    from accelerate.commands.accelerate_cli import main
  File "/usr/local/lib/python3.12/dist-packages/accelerate/__init__.py", line 16, in <module>
    from .accelerator import Accelerator
  File "/usr/local/lib/python3.12/dist-packages/accelerate/accelerator.py", line 34, in <module>
    from huggingface_hub import split_torch_state_dict_into_shards
ImportError: cannot import name 'split_torch_state_dict_into_shards' from 'huggingface_hub' (/usr/local/lib/python3.12/dist-packages/huggingface_hub/__init__.py)
15:42:39-048110 INFO     Saving v2-inference.yaml as                            
                         /content/drive/MyDrive/AI/loras/mvmw/lora_mvmw_20260214
                         -154205.yaml 

ImportError: cannot import name 'split_torch_state_dict_into_shards' from 'huggingface_hub' (/usr/local/lib/python3.12/dist-packages/huggingface_hub/init.py

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions