Hugging Face's logo

Spaces:
sagaxlearn
/
testing-groq
Sleeping

App Files Community
khurrameycon commited on
Commit
e6f1ab4
·
verified ·
1 Parent(s): d490720

model on GPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ model = MllamaForConditionalGeneration.from_pretrained(
33
  )
34
 
35
  # Move the model to the appropriate device (GPU if available)
36
- model.to(device)
37
  processor = AutoProcessor.from_pretrained(model_name, use_auth_token=HF_TOKEN)
38
 
39
  # @spaces.GPU # Use the free GPU provided by Hugging Face Spaces
 
33
  )
34
 
35
  # Move the model to the appropriate device (GPU if available)
36
+ # model.to(device)
37
  processor = AutoProcessor.from_pretrained(model_name, use_auth_token=HF_TOKEN)
38
 
39
  # @spaces.GPU # Use the free GPU provided by Hugging Face Spaces