Update README.md
Browse files
README.md
CHANGED
|
@@ -69,8 +69,7 @@ model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map='cuda:1', lo
|
|
| 69 |
tok = AutoTokenizer.from_pretrained(MODEL_NAME)
|
| 70 |
|
| 71 |
# Inference
|
| 72 |
-
prompt = "I want you to generate a theory that unites quantum mechanics with the theory of relativity and cosmic consciousness
|
| 73 |
-
"
|
| 74 |
inputs = tok([prompt], return_tensors="pt").to('cuda')
|
| 75 |
streamer = TextStreamer(tok)
|
| 76 |
|
|
|
|
| 69 |
tok = AutoTokenizer.from_pretrained(MODEL_NAME)
|
| 70 |
|
| 71 |
# Inference
|
| 72 |
+
prompt = "I want you to generate a theory that unites quantum mechanics with the theory of relativity and cosmic consciousness"
|
|
|
|
| 73 |
inputs = tok([prompt], return_tensors="pt").to('cuda')
|
| 74 |
streamer = TextStreamer(tok)
|
| 75 |
|