Tufan1 commited on
Commit
d5026a1
·
verified ·
1 Parent(s): 55b99fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -6
app.py CHANGED
@@ -30,12 +30,6 @@ model = PeftModel.from_pretrained(
30
  adapter_name="cardio_adapter"
31
  )
32
 
33
- # Explicitly load adapter weights to CPU
34
- adapter_weights = torch.load(
35
- f"{ADAPTER_PATH}/adapter_model.bin",
36
- map_location=torch.device('cpu')
37
- )
38
- model.load_state_dict(adapter_weights, strict=False)
39
 
40
  # Force CPU-safe model loading
41
  #base_model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, torch_dtype=torch.float32)
 
30
  adapter_name="cardio_adapter"
31
  )
32
 
 
 
 
 
 
 
33
 
34
  # Force CPU-safe model loading
35
  #base_model = AutoModelForCausalLM.from_pretrained(BASE_MODEL, torch_dtype=torch.float32)