Spaces:

KeerthiVM
/

SkinGPT

Sleeping

KeerthiVM commited on May 7

Commit

d09c700

1 Parent(s): 04bab96

fix added

Files changed (1) hide show

SkinGPT.py CHANGED Viewed

@@ -328,7 +328,7 @@ class SkinGPT4(nn.Module):
         full_output = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
-        # print(f"Full Output from llama : {full_output}")
         response = full_output.split("### Response:")[-1].strip()
         # print(f"Response from llama : {full_output}")

         full_output = self.tokenizer.decode(outputs[0], skip_special_tokens=True)
+        print(f"Full Output from llama : {full_output}")
         response = full_output.split("### Response:")[-1].strip()
         # print(f"Response from llama : {full_output}")