Spaces:

gnumanth
/

llama3-chat

Sleeping

gnumanth commited on Apr 22

Commit

bef7b02

•

1 Parent(s): 0c8a603

fix: import os

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
 device_map = 'cuda'

 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+import os
 model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
 device_map = 'cuda'