|
楼主 |
发表于 2023-2-24 02:12
|
显示全部楼层
本帖最后由 我輩樹である 于 2023-3-25 21:44 编辑
只是界面的限制,可以如下改一下:
gradio_pose2image.py文件的89行左右:
- image_resolution = gr.Slider(label="Image Resolution", minimum=256, maximum=<这里改成你想要的大小>, value=512, step=256)
复制代码
当然如果分辨率太大,可能会崩。
建议打开xformers加速。方法是用下面的environment.yaml构建环境。
仅测试了pose2image,如果你要用别的ControlNet功能慎用:
- name: control
- channels:
- - pytorch
- - nvidia
- - defaults
- dependencies:
- - python=3.8.10
- - pip
- - pytorch=1.13.1
- - torchvision
- - numpy=1.23.1
- - pytorch-cuda=11.7
- - pip:
- - -i https://pypi.tuna.tsinghua.edu.cn/simple
- - gradio==3.16.2
- - albumentations==1.3.0
- - opencv-contrib-python==4.3.0.36
- - imageio==2.9.0
- - imageio-ffmpeg==0.4.2
- - pytorch-lightning==1.5.0
- - omegaconf==2.1.1
- - test-tube>=0.7.5
- - streamlit==1.12.1
- - einops==0.3.0
- - transformers==4.19.2
- - webdataset==0.2.5
- - kornia==0.6
- - open_clip_torch==2.0.2
- - invisible-watermark>=0.1.5
- - streamlit-drawable-canvas==0.8.0
- - torchmetrics==0.6.0
- - timm==0.6.12
- - addict==2.4.0
- - yapf==0.32.0
- - prettytable==3.6.0
- - safetensors==0.2.7
- - basicsr==1.4.2
- - xformers
复制代码
|
|