vilarin commited on
Commit
10c9224
1 Parent(s): 17451fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -56,13 +56,14 @@ for filename in filenames:
56
  # })
57
 
58
  MODEL_PATH = os.path.abspath("model")
 
59
 
60
  model = AutoModelForCausalLM.from_pretrained(
61
- './model',
62
  torch_dtype=torch.float16,
63
  low_cpu_mem_usage=True,
64
  ).to(0)
65
- tokenizer = AutoTokenizer.from_pretrained('./model')
66
  vision_tower = model.get_vision_tower()
67
  vision_tower.load_model()
68
  vision_tower.to(device="cuda", dtype=torch.float16)
 
56
  # })
57
 
58
  MODEL_PATH = os.path.abspath("model")
59
+ print(MODEL_PATH)
60
 
61
  model = AutoModelForCausalLM.from_pretrained(
62
+ MODEL_PATH ,
63
  torch_dtype=torch.float16,
64
  low_cpu_mem_usage=True,
65
  ).to(0)
66
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH )
67
  vision_tower = model.get_vision_tower()
68
  vision_tower.load_model()
69
  vision_tower.to(device="cuda", dtype=torch.float16)