Spaces:
Runtime error
Runtime error
app.py
CHANGED
@@ -16,9 +16,9 @@ def model_init():
|
|
16 |
# device_map="auto",
|
17 |
# Ref. for `half`: Chan-Jan, Thanks!
|
18 |
).eval().to(device)
|
19 |
-
return
|
20 |
|
21 |
-
|
22 |
|
23 |
st.text(str(repr(model)))
|
24 |
|
|
|
16 |
# device_map="auto",
|
17 |
# Ref. for `half`: Chan-Jan, Thanks!
|
18 |
).eval().to(device)
|
19 |
+
return tokenizer, model
|
20 |
|
21 |
+
tokenizer, model = model_init()
|
22 |
|
23 |
st.text(str(repr(model)))
|
24 |
|