Kr08 commited on
Commit
510dbb2
·
verified ·
1 Parent(s): 6f80376

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
 
4
  from airllm import HuggingFaceModelLoader, AutoModelForCausalLM
5
 
6
- model_loader = HuggingFaceModelLoader("meta-llama/Meta-Llama-3-70B-Instruct")
7
  model = AutoModelForCausalLM.from_pretrained(model_loader)
8
 
9
  @spaces.GPU
 
3
 
4
  from airllm import HuggingFaceModelLoader, AutoModelForCausalLM
5
 
6
+ model_loader = HuggingFaceModelLoader("meta-llama/Meta-Llama-3-8B-Instruct")
7
  model = AutoModelForCausalLM.from_pretrained(model_loader)
8
 
9
  @spaces.GPU