Fung804 commited on
Commit
f46ba2f
β€’
1 Parent(s): 3a18326

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -4
app.py CHANGED
@@ -1,9 +1,21 @@
1
- import gradio as gr
2
- import tensorflow
3
- import os
 
 
 
 
 
 
4
  import torch
5
- from diffusers import StableDiffusionPipeline
 
6
  from torch import autocast
 
 
 
 
 
7
 
8
  def txt2img(prompt):
9
 
 
1
+ #!pip install --upgrade pip
2
+ !pip install diffusers==0.9.0
3
+ #it is still work for 0.9.0, if problem get, use below command
4
+ !pip install --upgrade diffusers transformers scipy
5
+ #!pip install transformers==4.24.0
6
+ #<--able go to no error
7
+
8
+ import random
9
+ import re
10
  import torch
11
+ import requests
12
+ from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
13
  from torch import autocast
14
+ import gradio
15
+ import gradio as gr
16
+
17
+ pipe = StableDiffusionPipeline.from_pretrained("Fung804/makoto-shinkai-v2", torch_dtype=torch.float16)
18
+ pipe = pipe.to("cuda")
19
 
20
  def txt2img(prompt):
21