Update inf.py
Browse files
inf.py
CHANGED
|
@@ -25,7 +25,7 @@ class InferencePipeline:
|
|
| 25 |
else:
|
| 26 |
self.pipe = StableDiffusionXLPipeline.from_pretrained(
|
| 27 |
self.base_model_id,
|
| 28 |
-
torch_dtype=torch.float16,
|
| 29 |
use_auth_token=self.hf_token)
|
| 30 |
self.pipe = self.pipe.to(self.device)
|
| 31 |
self.content_lora_model_id = None
|
|
|
|
| 25 |
else:
|
| 26 |
self.pipe = StableDiffusionXLPipeline.from_pretrained(
|
| 27 |
self.base_model_id,
|
| 28 |
+
# torch_dtype=torch.float16,
|
| 29 |
use_auth_token=self.hf_token)
|
| 30 |
self.pipe = self.pipe.to(self.device)
|
| 31 |
self.content_lora_model_id = None
|