Spaces:
Running
on
Zero
Running
on
Zero
layerdiffusion
commited on
Commit
•
93f83a7
1
Parent(s):
b3fef9e
- app.py +6 -0
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -1,3 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
|
| 3 |
def greet(name):
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
print(f"Is CUDA available: {torch.cuda.is_available()}")
|
| 3 |
+
# True
|
| 4 |
+
print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
|
| 5 |
+
# Tesla T4
|
| 6 |
+
|
| 7 |
import gradio as gr
|
| 8 |
|
| 9 |
def greet(name):
|
requirements.txt
CHANGED
|
@@ -1,10 +1,11 @@
|
|
|
|
|
|
|
|
| 1 |
diffusers==0.27.2
|
| 2 |
transformers==4.36.2
|
| 3 |
opencv-python
|
| 4 |
safetensors
|
| 5 |
pillow==10.2.0
|
| 6 |
einops
|
| 7 |
-
torch
|
| 8 |
peft
|
| 9 |
basicsr
|
| 10 |
gradio==3.41.2
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
+
torch
|
| 3 |
diffusers==0.27.2
|
| 4 |
transformers==4.36.2
|
| 5 |
opencv-python
|
| 6 |
safetensors
|
| 7 |
pillow==10.2.0
|
| 8 |
einops
|
|
|
|
| 9 |
peft
|
| 10 |
basicsr
|
| 11 |
gradio==3.41.2
|