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

相关推荐
武子康1 天前
不要问模型是 Transformer 还是 Diffusion:一套五层技术栈检查法(系统角色 / 表示空间 / 网络骨架 / 训练范式 / 推理算法)
人工智能·stable diffusion·llm
武子康3 天前
从随机动作块到真实闭环:Diffusion 与 Flow 策略的执行账本
人工智能·stable diffusion·agent
chenment4 天前
ComfyUI 自定义节点开发:从零扩展你的图像生成工作流
python·stable diffusion
Lifangyun_WD11 天前
RTX 5090跑Stable Diffusion XL:生图速度、显存占用与商业应用边界
人工智能·stable diffusion·gpu算力·rtx 5090·gpu容器·gpu租赁
承渊政道11 天前
把家里的显卡变成远程AI画室:Stable Diffusion WebUI部署实战
人工智能·stable diffusion·内网穿透·cpolar·web ui·ai画室
张彦峰ZYF19 天前
大模型LLM ACA - ACP认证考试真题冲刺演练二答案参考
人工智能·lora·llm·embedding·rag·few-shot
emfuture23 天前
中嵌科技“岷山“实验室技术拆解:基于 DeepSeek-7B 的工控垂类模型,怎么把推理压到 50ms 以内?
人工智能·科技·lora·知识图谱·工业大模型·垂类微调·工控ai
troubles maker2 个月前
LoRA: Low-Rank Adaptation of Large Language Models
语言模型·自然语言处理·lora·大模型·peft
老刘说AI2 个月前
类Sora模型:解锁动态视觉艺术的密码
人工智能·stable diffusion·架构·embedding
python-码博士2 个月前
Stable Diffusion 3 / FLUX 类模型中的 Flow Matching:怎么用、怎么调、怎么避坑
stable diffusion