今回の目標:取り敢えず画像を出力できるようになる。 Google Colab にアクセスする。 新規ノートを作成する。 右上の「接続」をクリック。接続されたら、 左上の「編集」内の「ノートブックの設定」を選択 ハードウェア アクセレータの項目をGPUに設定し「T4」に設定。保存を押す。 #有料ではあるが、 より高性能なGPUも選択できる。 すると、接続が一旦切れるので、再接続するのを確認する。 再生マークが実行、その隣にコードを打ち込む。 !nvidia-smi 実行して、GPUが使用されるようになっているか確認する。実行出来たら◯ 右上の「+コード」をクリックして新しい欄を追加。以下を記述。 !pip install --upgrade diffusers accelerate transformers 実行。 右上の「+コード」をクリックして新しい欄を追加。以下を記述。 from diffusers import StableDiffusionPipeline , DPMSolverMultistepScheduler import torch from diffusers import StableDiffusionPipeline # 利用したいAIモデル # Stable Diffusionにはさまざまな派生モデルがあります model_id = "gsdf/Counterfeit-V2.5" # パイプラインの作成 pipeline = StableDiffusionPipeline.from_pretrained ( model_id , torch_dtype=torch.float16 ) # GPUを使うように変更 pipeline = pipeline.to ( "cuda" ) 実行。 右上の「+コード」をクリックして新しい欄を追加。以下を記述。 prompt = """Unreal Engine, Realistic,photorealistic, 4k,8k,16k, elaborate, hyper quality, villa, glo...
コメント
コメントを投稿