Spaces:
Runtime error
Runtime error
Ubuntu
commited on
Commit
•
133dfc1
1
Parent(s):
c7f445d
Add application file
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
@@ -8,7 +8,7 @@ def produce_art(prompt):
|
|
8 |
import torch
|
9 |
|
10 |
model_id = "atsyplikhin/rita_sd_model"
|
11 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
12 |
|
13 |
bs = 1
|
14 |
images = pipe([prompt]*bs, num_inference_steps=50, guidance_scale=7.5)
|
8 |
import torch
|
9 |
|
10 |
model_id = "atsyplikhin/rita_sd_model"
|
11 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
|
12 |
|
13 |
bs = 1
|
14 |
images = pipe([prompt]*bs, num_inference_steps=50, guidance_scale=7.5)
|
requirements.txt
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
diffusers
|
2 |
gradio
|
3 |
torch
|
1 |
+
transformers
|
2 |
diffusers
|
3 |
gradio
|
4 |
torch
|