陆鹿 commited on
Commit
1937b85
1 Parent(s): 1906c3c

:sparkles: print cpu info

Browse files
Files changed (2) hide show
  1. app.py +2 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler
2
  import gradio as gr
 
3
  import torch
4
  from PIL import Image
5
  from diffusers import OnnxStableDiffusionPipeline
@@ -105,5 +106,6 @@ with gr.Blocks(css=css) as demo:
105
  </div>
106
  """)
107
 
 
108
  demo.queue(concurrency_count=1)
109
  demo.launch()
 
1
  from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DPMSolverMultistepScheduler
2
  import gradio as gr
3
+ import cpuinfo
4
  import torch
5
  from PIL import Image
6
  from diffusers import OnnxStableDiffusionPipeline
 
106
  </div>
107
  """)
108
 
109
+ print(cpuinfo.get_cpu_info())
110
  demo.queue(concurrency_count=1)
111
  demo.launch()
requirements.txt CHANGED
@@ -8,3 +8,4 @@ diffusers
8
  git+https://github.com/huggingface/transformers
9
  accelerate
10
  ftfy
 
 
8
  git+https://github.com/huggingface/transformers
9
  accelerate
10
  ftfy
11
+ py-cpuinfo