Spaces:
Running
on
Zero
Running
on
Zero
Jitesh Jain
commited on
Commit
·
346ff07
1
Parent(s):
f1653dd
:zap: Fix version
Browse files- app.py +2 -1
- requirements.txt +0 -4
app.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
import torch
|
| 3 |
import numpy as np
|
|
@@ -251,7 +252,7 @@ def regenerate(state, image_process_mode):
|
|
| 251 |
# @spaces.GPU
|
| 252 |
# def get_interm_outs(state):
|
| 253 |
|
| 254 |
-
|
| 255 |
@spaces.GPU
|
| 256 |
def generate(state, temperature, top_p, max_output_tokens, is_inter=False):
|
| 257 |
if is_inter:
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import gradio as gr
|
| 3 |
import torch
|
| 4 |
import numpy as np
|
|
|
|
| 252 |
# @spaces.GPU
|
| 253 |
# def get_interm_outs(state):
|
| 254 |
|
| 255 |
+
|
| 256 |
@spaces.GPU
|
| 257 |
def generate(state, temperature, top_p, max_output_tokens, is_inter=False):
|
| 258 |
if is_inter:
|
requirements.txt
CHANGED
|
@@ -25,10 +25,6 @@ diffusers===0.27.2
|
|
| 25 |
protobuf
|
| 26 |
accelerate==0.27.2
|
| 27 |
scikit-learn
|
| 28 |
-
icecream
|
| 29 |
-
datasets
|
| 30 |
-
pytorch-fid
|
| 31 |
-
lpips
|
| 32 |
opencv-python-headless
|
| 33 |
huggingface_hub==0.24.7
|
| 34 |
transformers==4.41.1
|
|
|
|
| 25 |
protobuf
|
| 26 |
accelerate==0.27.2
|
| 27 |
scikit-learn
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
opencv-python-headless
|
| 29 |
huggingface_hub==0.24.7
|
| 30 |
transformers==4.41.1
|