Spaces:
Sleeping
Sleeping
tokeron
commited on
Commit
•
d55e7d8
1
Parent(s):
2ec2ebd
update torch
Browse files- app.py +1 -1
- requirements.txt +3 -0
app.py
CHANGED
|
@@ -47,7 +47,7 @@ def get_prompt(prompt):
|
|
| 47 |
# print('number of images:', len(images))
|
| 48 |
# images = prepare_images(images)
|
| 49 |
# print('done preparing images')
|
| 50 |
-
image = get_images(prompt)
|
| 51 |
return image
|
| 52 |
|
| 53 |
if __name__ == '__main__':
|
|
|
|
| 47 |
# print('number of images:', len(images))
|
| 48 |
# images = prepare_images(images)
|
| 49 |
# print('done preparing images')
|
| 50 |
+
image = get_images(prompt, skip_layers=0)
|
| 51 |
return image
|
| 52 |
|
| 53 |
if __name__ == '__main__':
|
requirements.txt
CHANGED
|
@@ -1,3 +1,6 @@
|
|
| 1 |
transformers
|
| 2 |
diffusers
|
| 3 |
gradio
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
transformers
|
| 2 |
diffusers
|
| 3 |
gradio
|
| 4 |
+
torch
|
| 5 |
+
torchvision
|
| 6 |
+
torchtext
|