AIGC - Stable Diffusion 搭建【从零到一】

AIGC - Stable Diffusion 搭建

1. Huggingface Token

url = ’‘
1. 注册获取账号
2. 生成获取用户项目token

2. Cloud GPU

url = '' # 可租用矩池云
1. 注册获取账号
2. 选择GPU档次及预装型号（建议huggingface,Pytorch）
3. huggingface-cli login  # Cloud服务器通过token验证 | 下载模型 
4. JupyterLab 
   pip3 install diffusers 
   pip3 install accelerate
   pip3 install spacy ftfy==4.4.3
   

3. autocover.txt

import diffusers, torch, os
from time import strftime, localtime

model_id = "CompVis/stable-diffusion-v1-4"
device = torch.device("cuda")
pipe = diffusers.StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
pipe = pipe.to(device)
image = pipe(prompt).images[0]
timeprint = strftime("%Y%m%d%H%M%S", localtime())

fldpath = os.path.join('/mnt/', 'xxx/img')
if not os.path.exists(fldpath):
    os.mkdir(fldpath)
img_filenm = f'{timeprint}.png'
image.save(os.path.join(fldpath, img_filenm))
print(f'{timeprint}.png')

4. userinput_write_pyfile.py:

def userinput_write_pyfile(user_input,local_put_path):
	'''
	- 将用户输入的文字信息写入远程py待执行文件中
	'''
    import os
    f = open(local_put_path,'r')
    index = 0
    lines = []
    for line in f.readlines():
        index +=1
        if index == 3:
            prompt = f'prompt = "{
     
     user_input}"'
            lines.append(prompt)
        lines.append(line)
    usr_path = os.path.join(os.path.expanduser('~'), 'Desktop/SSH/upload')
    pyfilepath = os.path.join(usr_path,'autocover.py')
    pf = open(pyfilepath,'w')
    for line in lines:
        pf.write(''.join(line))
    pf.close()
    return pyfilepath

5. SSH API

# -*- coding:utf-8 -*-
def get_SSH_account():
    host         =  ""            # 远程服务器ip
    port         =  0000          # 远程服务器端口
    password     =  ""            # 远程服务器密码
    user         =  ""            # 远程服务器用户名
    account_info = [host,port,password,user]    # 返回SSH登陆信息
    return account_info
    
def SSH_connection(user_input,accounts,paths,write_py, specail_fname,is_DownloadALL,filter_fname):
    import paramiko,os,time
    (host, port, password, user) = accounts
    (remote_get_path, remote_put_paths, local_get_path, local_put_paths, pyfilepath, python_path) = paths

    class Pysftp(object):
        """
        python3 自动上传、下载文件
        """
        global local_put_paths, local_put_path, local_get_path, remote_put_paths,remote_put_path, remote_get_path, pyfilepath, python_path

        def __init__(self, host, port, user, password):
            self.host = host
            self.port = port
            self.user = user
            self.password = password

        def connect(self):
            """
            建立ssh连接
            """
            self.ssh = paramiko.SSHClient()
            self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            self.ssh.connect(self.host, self.port, self.user, self.password)
            print("[SSH] -> 连接成功")

        # def cmd (self, cmd):
        #     """