Spaces:
Build error
Build error
Commit
·
751b158
1
Parent(s):
84974ef
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ from rudolph.pipelines import generate_codebooks, self_reranking_by_image, self_
|
|
| 9 |
from rudolph.pipelines import zs_clf
|
| 10 |
import gradio as gr
|
| 11 |
from rudolph import utils
|
| 12 |
-
device = '
|
| 13 |
|
| 14 |
model = get_rudolph_model('350M', fp16=True, device=device)
|
| 15 |
model.load_state_dict(torch.load("awesomemodel__dalle_1500.pt"))
|
|
|
|
| 9 |
from rudolph.pipelines import zs_clf
|
| 10 |
import gradio as gr
|
| 11 |
from rudolph import utils
|
| 12 |
+
device = 'cpu'
|
| 13 |
|
| 14 |
model = get_rudolph_model('350M', fp16=True, device=device)
|
| 15 |
model.load_state_dict(torch.load("awesomemodel__dalle_1500.pt"))
|