Hugging Face's logo

Spaces:
johann22
/
idefics-9b-ft-describe-diffusion-mj
Runtime error

App Files Community
johann22 commited on
Commit
acacdc1
·
1 Parent(s): 66b4b64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ client = InferenceClient("HuggingFaceM4/idefics-9b-instruct")
10
 
11
  #peft_model_id = "mrm8488/idefics-9b-ft-describe-diffusion-bf16-adapter"
12
  peft_model_id = "HuggingFaceM4/idefics-9b"
13
- #device = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
 
16
  #config = PeftConfig.from_pretrained(peft_model_id)
 
10
 
11
  #peft_model_id = "mrm8488/idefics-9b-ft-describe-diffusion-bf16-adapter"
12
  peft_model_id = "HuggingFaceM4/idefics-9b"
13
+ device = "cuda" if torch.cuda.is_available() else "cpu"
14
 
15
 
16
  #config = PeftConfig.from_pretrained(peft_model_id)