Spaces:

dx2102
/

llama-midi

Running on Zero

dx2102 commited on Jan 26

Commit

cd71c82

verified ·

1 Parent(s): 8087323

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ print('Loading model...')
 # pipe = transformers.pipeline("text-generation", model="openai-community/gpt2")
 pipe = transformers.pipeline(
     "text-generation",
-    model="dx2102/llama-midi",
     torch_dtype="bfloat16",
     device_map="auto",
 )

 # pipe = transformers.pipeline("text-generation", model="openai-community/gpt2")
 pipe = transformers.pipeline(
     "text-generation",
+    model="dx2102/llama-midi-3B",
     torch_dtype="bfloat16",
     device_map="auto",
 )