ironjr commited on
Commit
0f05e84
·
verified ·
1 Parent(s): 82a8f22

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -379,7 +379,7 @@ class StreamMultiDiffusion(nn.Module):
379
  A single string of text prompt.
380
  """
381
  question = 'Question: What are in the image? Answer:'
382
- print(self.i2t_processor.device, self.i2t_model.device)
383
  inputs = self.i2t_processor(image, question, return_tensors='pt')
384
  print(inputs)
385
  out = self.i2t_model.generate(**inputs, max_new_tokens=77)
 
379
  A single string of text prompt.
380
  """
381
  question = 'Question: What are in the image? Answer:'
382
+ print(self.i2t_model.device)
383
  inputs = self.i2t_processor(image, question, return_tensors='pt')
384
  print(inputs)
385
  out = self.i2t_model.generate(**inputs, max_new_tokens=77)