facechain环境部署

环境安装

powershell 复制代码
# 创建虚拟环境facechain
conda create -n facechain python=3.8
conda activate facechain
# 克隆
GIT_LFS_SKIP_SMUDGE=1 git clone https://github.com/modelscope/facechain.git --depth 1
# 安装第三方库
cd facechain
pip install -r requirements.txt
pip install -U openmim
# 设置所需的CUDA,因为安装了多个版本的CUDA
export PATH=/home/xxx/.local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=LD_LIBRARY_PATH:/home/xxx/.local/cuda-11.7/lib64
export CUDA_HOME=/home/xxx/.local/cuda-11.7
# 安装mmcv
mim install mmcv-full==1.7.0

webui的运行方式

  • app.py文件增加临时文件存放位置,因为多个人使用同一个服务,gradio的临时文件会存放在/tmp/gradio下存在权限问题。

    python 复制代码
     import tempfile
     tempfile.tempdir = './tmp'
  • app.py文件修改modelscope和huggingface的目录

    python 复制代码
    	import os
    	os.environ['MODELSCOPE_CACHE'] = '/opt/buty/work/modelscope'
    	os.environ['HF_HOME'] = '/opt/buty/work/huggingface'
    	os.environ['CUDA_VISIBLE_DEVICES'] = '0'
  • snapdown 文件的路径使用临时文件存放位置还是有问题,需要修改/tmp/snapdown/目录的权限

    powershell 复制代码
    sudo chmod -R 777 /tmp/snapdown/
  • 运行脚本

    powershell 复制代码
        python app.py

命令行交互界面的运行方式

  • train_text_to_image_lora.py文件修改modelscope和huggingface的目录

    python 复制代码
    	import os
    	os.environ['MODELSCOPE_CACHE'] = '/opt/buty/work/modelscope'
    	os.environ['HF_HOME'] = '/opt/buty/work/huggingface'
    	os.environ['CUDA_VISIBLE_DEVICES'] = '0'
  • 模型训练

    python 复制代码
    PYTHONPATH=. sh train_lora.sh "ly261666/cv_portrait_model" "v2.0" "film/film" "./imgs" "./processed" "./output"
  • 推理

    python 复制代码
    python run_inference.py

pycharm的运行方式

复制代码
train_text_to_image_lora.py脚本Edit Configurations设置:
Name:accelerate.commands.launch
Run:facechain module accelerate.commands.launch
Parameters:
		facechain/train_text_to_image_lora.py
		--pretrained_model_name=ly261666/cv_portrait_model
		--revision=v2.0
		--sub_path=film/film
		--train_data_dir=./imgs
		--output_dataset_name=./processed
		--output_dir=./output
Working directory:/opt/buty/work/facechain
Environment variables:
        CUDA_VISIBLE_DEVICES:0
		PYTHONPATH: .

参考资料
modelscope/facechain

相关推荐
愚公搬代码3 小时前
【愚公系列】《AI绘画:Stable Diffusion ComfyUl的艺术》007-共享WebUI 的路径
ai作画·stable diffusion
墨言智语13 小时前
003、方法一进阶:AI绘画与短视频制作——用Midjourney/Stable Diffusion接单赚钱
ai作画·stable diffusion·midjourney
愚公搬代码14 小时前
【愚公系列】《AI绘画:Stable Diffusion ComfyUl的艺术》008-ComfyUI整合包
ai作画·stable diffusion
OpalStag581 天前
MusePublic Art Studio多场景:短视频封面/直播背景/AR滤镜素材生成
stable diffusion·短视频制作·内容创作·ai图像生成
日光明媚1 天前
torch.compile 与 Triton 的加速本质:从原理到实际效果
人工智能·python·计算机视觉·stable diffusion·aigc
小贺儿开发3 天前
Unity3D 本地 Stable Diffusion 文生图效果演示
人工智能·unity·stable diffusion·文生图·ai绘画·本地化
Java后端的Ai之路3 天前
什么是“多模态微调”?
人工智能·lora·微调
Flying pigs~~4 天前
LoRA 面试完全指南:低秩分解原理 + Transformer 应用
人工智能·深度学习·lora·大模型·微调·transformer
Anesthesia丶8 天前
Qwen2.5-1.5b 模型部署与LORA训练笔记
pytorch·python·lora·llm·qwen·vllm
m0_6501082410 天前
LoRA:大语言模型低秩适配
论文阅读·lora·大模型微调·经典方法