Aashi commited on
Commit
38549eb
1 Parent(s): d67efb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -21,8 +21,6 @@ def load_llama_model():
21
  device_map="auto",
22
  # offload_folder="offload",
23
  ).to("cuda")
24
-
25
- )
26
  processor = AutoProcessor.from_pretrained(model_id)
27
 
28
  return model, processor
 
21
  device_map="auto",
22
  # offload_folder="offload",
23
  ).to("cuda")
 
 
24
  processor = AutoProcessor.from_pretrained(model_id)
25
 
26
  return model, processor