Yardenfren commited on
Commit
9bc5970
·
verified ·
1 Parent(s): 4c3e795

Update inf.py

Browse files
Files changed (1) hide show
  1. inf.py +1 -1
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