Ifeanyi commited on
Commit
634e30c
·
1 Parent(s): 154c1fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -0
app.py CHANGED
@@ -2,6 +2,16 @@ from diffusers import StableDiffusionXLPipeline
2
  import gradio as gr
3
  import torch
4
 
 
 
 
 
 
 
 
 
 
 
5
  def segMindImage(prompt, negative_prompt):
6
  pipe = StableDiffusionXLPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
7
  pipe.to("cuda")
 
2
  import gradio as gr
3
  import torch
4
 
5
+ #from transformers.utils.hub import move_cache
6
+ import pkg_resources
7
+ #move_cache()
8
+
9
+ print("------------------------")
10
+ package_name = "transformers"
11
+ version = pkg_resources.get_distribution(package_name).version
12
+ print(f"The version of {package_name} is: {version}")
13
+ print("------------------------")
14
+
15
  def segMindImage(prompt, negative_prompt):
16
  pipe = StableDiffusionXLPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
17
  pipe.to("cuda")