Spaces:
Runtime error
Runtime error
Update model.py
Browse files
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.
|
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)
|